Commit 3e3bd99a authored by Frisinghelli Daniel's avatar Frisinghelli Daniel
Browse files

Merge branch 'optim' into ai4ebv-public

parents 64e94ac1 bd875371
......@@ -1295,6 +1295,8 @@ class NetworkTrainer(BaseConfig):
# decay learning rate, if scheduler is specified
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()))
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