Skip to content

Commit 84b9a12

Browse files
committed
Cleaner way to get options.
1 parent 5863345 commit 84b9a12

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

gluefactory/train.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -153,11 +153,11 @@ def get_lr_scheduler(optimizer, conf):
153153
scheduler = get_lr_scheduler(optimizer, scheduler_conf)
154154
schedulers.append(scheduler)
155155

156-
# remove conf.options.schedulers
157-
del conf.options.schedulers
156+
options = {k: v for k, v in conf.options.items() if k != "schedulers"}
158157
return getattr(torch.optim.lr_scheduler, conf.type)(
159-
optimizer, schedulers, **conf.options
158+
optimizer, schedulers, **options
160159
)
160+
161161
return getattr(torch.optim.lr_scheduler, conf.type)(optimizer, **conf.options)
162162

163163
# backward compatibility

0 commit comments

Comments
 (0)