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

add valuerror

This commit is contained in:
shahules786 2022-09-14 11:51:03 +05:30
parent c5fdfbe188
commit 042ce2b525
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/utils/loss.py
View File

@ -11,7 +11,7 @@ class mean_squared_error(nn.Module):
self.loss_fun = nn.MSELoss(reduction=reduction) self.loss_fun = nn.MSELoss(reduction=reduction)
def forward(self,prediction:torch.Tensor, target: torch.Tensor): def forward(self,prediction:torch.Tensor, target: torch.Tensor):
if prediction.size() != target.size() or target.ndim < 3: if prediction.size() != target.size() or target.ndim < 3:
raise TypeError(f"""Inputs must be of the same shape (batch_size,channels,samples) raise TypeError(f"""Inputs must be of the same shape (batch_size,channels,samples)
got {prediction.size()} and {target.size()} instead""") got {prediction.size()} and {target.size()} instead""")
@ -46,7 +46,7 @@ class Avergeloss(nn.Module):
def validate_loss(self,loss:str): def validate_loss(self,loss:str):
if loss not in LOSS_MAP.keys(): if loss not in LOSS_MAP.keys():
raise ValueError() raise ValueError(f"Invalid loss function {loss}, available loss functions are {LOSS_MAP.keys()}")
else: else:
return LOSS_MAP[loss] return LOSS_MAP[loss]