LearningRateMonitor(logging_interval='epoch',...) should plot learning-rate versus epochs as opposed to learning-rate versus steps
🚀 Feature
LearningRateMonitor(logging_interval='epoch',...) should plot learning-rate versus epochs as opposed to learning-rate versus steps
Motivation
Since I am specifying that the logging-interval is in epochs, I expect TensorBoard to show the x-axis in epochs as opposed to steps.
Additional context
I have made the following change in line 131 of file pytorch_lightning/callbacks/lr_monitor.py:
Replace:
trainer.logger.log_metrics(latest_stat, step=trainer.global_step)
with:
trainer.logger.log_metrics(latest_stat, step=trainer.current_epoch)
This change in code works. I am using Pytorch-Lighting version 1.1.4