File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change @@ -15,8 +15,8 @@ torch-optimizer
15
15
:target: https://deepsource.io/gh/jettify/pytorch-optimizer/?ref=repository-badge
16
16
17
17
18
- **torch-optimizer ** -- collection of optimizers for PyTorch _.
19
-
18
+ **torch-optimizer ** -- collection of optimizers for PyTorch _ compatible with optim _
19
+ module.
20
20
21
21
22
22
Simple example
@@ -38,6 +38,11 @@ Installation process is simple, just::
38
38
$ pip install torch_optimizer
39
39
40
40
41
+ Documentation
42
+ -------------
43
+ https://pytorch-optimizer.rtfd.io
44
+
45
+
41
46
Supported Optimizers
42
47
====================
43
48
@@ -643,3 +648,4 @@ SGD (PyTorch built-in)
643
648
.. _Rastrigin : https://en.wikipedia.org/wiki/Rastrigin_function
644
649
.. _Rosenbrock : https://en.wikipedia.org/wiki/Rosenbrock_function
645
650
.. _benchmark : https://en.wikipedia.org/wiki/Test_functions_for_optimization
651
+ .. _optim : https://pytorch.org/docs/stable/optim.html
You can’t perform that action at this time.
0 commit comments