Skip to content

Commit c09d18b

Browse files
authored
Merge pull request #353 from kozistr/update/schedulefree-optimizers
[Update] ScheduleFree optimizers
2 parents d18bb4b + dc08ba7 commit c09d18b

18 files changed

+74
-84
lines changed

docs/changelogs/v3.4.2.md

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,20 @@
55
* Implement `SCION` optimizer. (#348, #352)
66
* [Training Deep Learning Models with Norm-Constrained LMOs](https://arxiv.org/abs/2502.07529)
77

8+
### Update
9+
10+
* Update ScheduleFreeSGD, AdamW, RAdam optimizers with the latest. (#351, #353)
11+
* Remove `use_palm` variant in ScheduleFree optimizer due to instability. (#353)
12+
* Ranger25 optimizer. (#353)
13+
14+
### Fix
15+
16+
* Remove `weight decouple` parameter in ScheduleFree optimizers. (#351, #353)
17+
818
### Docs
919

1020
* Fix `AliG` optimizer visualization. (#350)
1121

1222
### Contributions
1323

14-
thanks to @AidinHamedi
24+
thanks to @AidinHamedi, @hatonosuke

docs/visualization.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -302,6 +302,10 @@
302302

303303
![image](https://raw.githubusercontent.com/kozistr/pytorch_optimizer/main/docs/visualizations/rastrigin_ScheduleFreeSGD.png)
304304

305+
### SCION
306+
307+
![image](https://raw.githubusercontent.com/kozistr/pytorch_optimizer/main/docs/visualizations/rastrigin_SCION.png)
308+
305309
### SGD
306310

307311
![image](https://raw.githubusercontent.com/kozistr/pytorch_optimizer/main/docs/visualizations/rastrigin_SGD.png)
@@ -668,6 +672,10 @@
668672

669673
![image](https://raw.githubusercontent.com/kozistr/pytorch_optimizer/main/docs/visualizations/rosenbrock_ScheduleFreeSGD.png)
670674

675+
### SCION
676+
677+
![image](https://raw.githubusercontent.com/kozistr/pytorch_optimizer/main/docs/visualizations/rosenbrock_SCION.png)
678+
671679
### SGD
672680

673681
![image](https://raw.githubusercontent.com/kozistr/pytorch_optimizer/main/docs/visualizations/rosenbrock_SGD.png)
8.83 KB
Loading
632 KB
Loading
90 Bytes
Loading
277 Bytes
Loading
-977 Bytes
Loading
9.31 KB
Loading
144 KB
Loading
2.57 KB
Loading

0 commit comments

Comments
 (0)