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 74caee6 commit 7c17b14Copy full SHA for 7c17b14
hubconf.py
@@ -1,10 +1,15 @@
1
dependencies = ["torch"]
2
3
-from functools import partial
+from functools import partial as _partial, update_wrapper as _update_wrapper
4
5
-from pytorch_optimizer import get_supported_optimizers, load_optimizer
+from pytorch_optimizer import (
6
+ get_supported_optimizers as _get_supported_optimizers,
7
+ load_optimizer as _load_optimizer,
8
+)
9
-for optimizer in get_supported_optimizers():
10
+for optimizer in _get_supported_optimizers():
11
name = optimizer.__name__
12
for n in (name, name.lower()):
- globals()[n] = partial(load_optimizer, optimizer=n)
13
+ func = _partial(_load_optimizer, optimizer=n)
14
+ _update_wrapper(func, optimizer)
15
+ globals()[n] = func
0 commit comments