diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-04 22:58:07 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-04 22:58:07 +0200 |
| commit | 969e1d5e179d9c42ffae0c9b12c9bd3be6091360 (patch) | |
| tree | ab849c38bc9b863afad85fd04d6f618031000e6f /training/conf/callbacks/learning_rate_monitor.yaml | |
| parent | 4da7a2c812221d56a430b35139ac40b23fa76f77 (diff) | |
Add wandb callbacks
Diffstat (limited to 'training/conf/callbacks/learning_rate_monitor.yaml')
| -rw-r--r-- | training/conf/callbacks/learning_rate_monitor.yaml | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/training/conf/callbacks/learning_rate_monitor.yaml b/training/conf/callbacks/learning_rate_monitor.yaml index 11a5ecf..4a14e1f 100644 --- a/training/conf/callbacks/learning_rate_monitor.yaml +++ b/training/conf/callbacks/learning_rate_monitor.yaml @@ -1,4 +1,4 @@ learning_rate_monitor: - type: LearningRateMonitor - args: - logging_interval: step + _target_: pytorch_lightning.callbacks.LearningRateMonitor + logging_interval: step + log_momentum: false |