[sgd] Avoid parameter "gotcha" for learning rate scheduler (#8107)

* with-scheduler-creator

* none

* add_freq

* runner

* torch
This commit is contained in:
Richard Liaw
2020-04-21 01:01:04 -07:00
committed by GitHub
parent d15609ba2a
commit fa7eecf48a
5 changed files with 44 additions and 24 deletions
+1 -1
View File
@@ -44,7 +44,7 @@ class TorchRunner:
use_fp16=False,
use_tqdm=False,
apex_args=None,
scheduler_step_freq="batch"):
scheduler_step_freq=None):
self.model_creator = model_creator
self.optimizer_creator = optimizer_creator
self.loss_creator = loss_creator