geekyrakshit commited on
Commit
ba4ac14
·
1 Parent(s): d7c18de

bug fix in training

Browse files
Files changed (1) hide show
  1. enhance_me/zero_dce/zero_dce.py +1 -1
enhance_me/zero_dce/zero_dce.py CHANGED
@@ -149,7 +149,7 @@ class ZeroDCE(Model):
149
  ]
150
  if self.using_wandb:
151
  callbacks += [WandbCallback()]
152
- history = self.model.fit(
153
  self.train_dataset,
154
  validation_data=self.val_dataset,
155
  epochs=epochs,
 
149
  ]
150
  if self.using_wandb:
151
  callbacks += [WandbCallback()]
152
+ history = self.fit(
153
  self.train_dataset,
154
  validation_data=self.val_dataset,
155
  epochs=epochs,