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 f337814 commit becf7cdCopy full SHA for becf7cd
hubconf.py
@@ -17,13 +17,13 @@
17
for _optimizer in _get_supported_optimizers():
18
name: str = _optimizer.__name__
19
_func = _partial(_load_optimizer, optimizer=name)
20
- _update_wrapper(_func, _optimizer)
+ _update_wrapper(_func, _optimizer.__init__)
21
for n in (name, name.lower(), name.upper()):
22
globals()[n] = _func
23
24
for _scheduler in _get_supported_lr_schedulers():
25
name: str = _scheduler.__name__
26
_func = _partial(_load_lr_scheduler, lr_scheduler=name)
27
- _update_wrapper(_func, _scheduler)
+ _update_wrapper(_func, _scheduler.__init__)
28
29
0 commit comments