Commit b96d568d authored by Frisinghelli Daniel's avatar Frisinghelli Daniel
Browse files

Merge branch 'optim' into ai4ebv-public

parents 3e3bd99a 6f69c807
......@@ -1296,7 +1296,7 @@ class NetworkTrainer(BaseConfig):
if self.lr_scheduler is not None:
self.lr_scheduler.step()'Epoch: {:d}, Learning rate: {:.5f}'.format(
epoch + 1, self.lr_scheduler.get_last_lr()))
epoch + 1, self.lr_scheduler.get_last_lr().pop()))
return self.training_state
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment