Spaces:

qgyd2021
/

nx_denoise

Running

HoneyTian commited on 8 days ago

Commit

0e1473e

1 Parent(s): 83f4e0f

update

Files changed (1) hide show

examples/frcrn/step_2_train_model.py CHANGED Viewed

@@ -183,7 +183,7 @@ def main():
         step_idx = int(step_idx)
         if step_idx > last_step_idx:
             last_step_idx = step_idx
-    last_epoch = 1
     if last_step_idx != -1:
         logger.info(f"resume from steps-{last_step_idx}.")

         step_idx = int(step_idx)
         if step_idx > last_step_idx:
             last_step_idx = step_idx
+    last_epoch = 0
     if last_step_idx != -1:
         logger.info(f"resume from steps-{last_step_idx}.")