Spaces:

geekyrakshit
/

enhance-me

Runtime error

geekyrakshit commited on Nov 28, 2021

Commit

ba4ac14

1 Parent(s): d7c18de

bug fix in training

Files changed (1) hide show

enhance_me/zero_dce/zero_dce.py CHANGED Viewed

@@ -149,7 +149,7 @@ class ZeroDCE(Model):
         ]
         if self.using_wandb:
             callbacks += [WandbCallback()]
-        history = self.model.fit(
             self.train_dataset,
             validation_data=self.val_dataset,
             epochs=epochs,

         ]
         if self.using_wandb:
             callbacks += [WandbCallback()]
+        history = self.fit(
             self.train_dataset,
             validation_data=self.val_dataset,
             epochs=epochs,