some more tests for model saving

2026-09-12 12:11:47 +08:00 · 2018-02-21 07:21:44 -08:00
parent 6efb761139
commit d510f0e8aa
2 changed files with 2 additions and 3 deletions
@@ -218,8 +218,7 @@ def main(args):
                if c.checkpoint:
                    # save model
                    save_checkpoint(model, optimizer, linear_loss.data[0],
-                                    best_loss, OUT_PATH,
-                                    current_step, epoch)
+                                    OUT_PATH, current_step, epoch)

                # Diagnostic visualizations
                const_spec = linear_output[0].data.cpu().numpy()