From 124ee8318064932b702fbb23bd3c3246a5c5d31f Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Thu, 6 Oct 2022 11:30:00 +0530
Subject: [PATCH] fix setter

---
 enhancer/models/model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/enhancer/models/model.py b/enhancer/models/model.py
index 39dbe80..6e6b4e1 100644
--- a/enhancer/models/model.py
+++ b/enhancer/models/model.py
@@ -74,9 +74,9 @@ class Model(pl.LightningModule):
     def loss(self, loss):
 
         if isinstance(loss, str):
-            losses = [loss]
+            loss = [loss]
 
-        self._loss = Avergeloss(losses)
+        self._loss = Avergeloss(loss)
 
     @property
     def metric(self):