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

View File

@ -46,7 +46,7 @@ class Avergeloss(nn.Module):
def validate_loss(self,loss:str):
if loss not in LOSS_MAP.keys():
raise ValueError()
raise ValueError(f"Invalid loss function {loss}, available loss functions are {LOSS_MAP.keys()}")
else:
return LOSS_MAP[loss]