We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5863345 commit 84b9a12Copy full SHA for 84b9a12
gluefactory/train.py
@@ -153,11 +153,11 @@ def get_lr_scheduler(optimizer, conf):
153
scheduler = get_lr_scheduler(optimizer, scheduler_conf)
154
schedulers.append(scheduler)
155
156
- # remove conf.options.schedulers
157
- del conf.options.schedulers
+ options = {k: v for k, v in conf.options.items() if k != "schedulers"}
158
return getattr(torch.optim.lr_scheduler, conf.type)(
159
- optimizer, schedulers, **conf.options
+ optimizer, schedulers, **options
160
)
+
161
return getattr(torch.optim.lr_scheduler, conf.type)(optimizer, **conf.options)
162
163
# backward compatibility
0 commit comments