Qian Zhao b1b1f53db2

[Fix] Fix optim_wrapper unittest for pytorch <= 1.10.0 (#975)

2023-03-02 14:14:23 +08:00

..

test_optimizer_wrapper_dict.py

[Refactor] Refactor the accumulate gradient implemention of OptimWrapper (#284)

2022-06-13 23:20:53 +08:00

test_optimizer_wrapper.py

[Fix] Fix optim_wrapper unittest for pytorch <= 1.10.0 (#975)

2023-03-02 14:14:23 +08:00

test_optimizer.py

[Feature] Add Lion optimizer (#952)

2023-02-23 11:24:50 +08:00