Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix typo

This commit is contained in:
shahules786 2022-09-27 15:52:51 +05:30
parent e4db841ebb
commit 634f146ca7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
enhancer/loss.py
View File

@ -45,7 +45,7 @@ class Avergeloss(nn.Module):
def validate_loss(self,loss:str):
if loss not in LOSS_MAP.keys():
raise ValueError(f"Invalid loss function {loss}, available loss functions are {LOSS_MAP.keys()}")
raise ValueError(f"Invalid loss function {loss}, available loss functions are {tuple([loss for loss in LOSS_MAP.keys()])}")
else:
return LOSS_MAP[loss]
@ -59,6 +59,6 @@ class Avergeloss(nn.Module):
LOSS_MAP = {"mea":mean_absolute_error, "mse": mean_squared_error}
LOSS_MAP = {"mae":mean_absolute_error, "mse": mean_squared_error}