Commit 6f69c807 authored by Frisinghelli Daniel's avatar Frisinghelli Daniel
Browse files

Log learning rate.

parent bd875371
......@@ -1296,7 +1296,7 @@ class NetworkTrainer(BaseConfig):
if self.lr_scheduler is not None:
self.lr_scheduler.step()
LOGGER.info('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