Skip to content
Snippets Groups Projects
Commit 96e9d37b authored by Tamino Huxohl's avatar Tamino Huxohl
Browse files

fix usage of snapshot epoch param in training

parent aadd524d
No related branches found
No related tags found
No related merge requests found
......@@ -40,7 +40,7 @@ class Training():
self.lr_scheduler.step()
logger.debug(f"Update learning rate from {_previous:.4f} to {self.lr_scheduler.get_last_lr()[0]:.4f}")
if epoch % self.snapshot_epoch:
if epoch % self.snapshot_epoch == 0:
self.store_snapshot(epoch)
logger.debug(f"Finished epoch {str(epoch):>{len(str(self.epochs))}}/{self.epochs + 1}")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment