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 251c5ce commit 4c4b26bCopy full SHA for 4c4b26b
README.md
@@ -46,8 +46,13 @@ optimizer = AdamP(model.parameters())
46
47
from pytorch_optimizer import load_optimizer
48
49
-model = YourModel()
50
-opt = load_optimizer(optimizer='adamp')
+optimizer = load_optimizer(optimizer='adamp')(model.parameters())
+
51
+# if you install `bitsandbytes` optimizer, you can use `8-bit` optimizers from `pytorch-optimizer`.
52
53
+from pytorch_optimizer import load_optimizer
54
55
+opt = load_optimizer(optimizer='bnb_adamw8bit')
56
optimizer = opt(model.parameters())
57
```
58
0 commit comments