Bug fix.

DavidBoja · DavidBoja · commit 26134918f291 · 2024-08-23T18:06:15.000Z
diff --git a/losses.py b/losses.py
@@ -171,7 +171,7 @@ def __init__(self, cfg: dict, loss_weights: dict,**kwargs):
 
         self.loss_fns = {}
         for loss_name in self.loss_names:
-            loss_fn = eval(LOSS_MAPPER[loss_name])(cfg) 
+            loss_fn = eval(LOSS_MAPPER[loss_name])(**cfg) 
             self.loss_fns[loss_name] = loss_fn
     
 
