add valuerror
This commit is contained in:
parent
c5fdfbe188
commit
042ce2b525
|
|
@ -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]
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue