mirror of
https://github.com/wassname/ray.git
synced 2026-08-10 12:30:14 +08:00
[sgd] Avoid parameter "gotcha" for learning rate scheduler (#8107)
* with-scheduler-creator * none * add_freq * runner * torch
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user