From 634f146ca713ef825abaec62b810aa08645654ff Mon Sep 17 00:00:00 2001 From: shahules786 Date: Tue, 27 Sep 2022 15:52:51 +0530 Subject: [PATCH] fix typo --- enhancer/loss.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/enhancer/loss.py b/enhancer/loss.py index af6a4ba..693e610 100644 --- a/enhancer/loss.py +++ b/enhancer/loss.py @@ -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}