ColossalAI/tests/test_optimizer/test_cpu_adam.py at 1cf6d92d7c93a26e29cadeb71bb34ee96b149a28 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-06 00:08:09 +00:00

Files

Jiarui Fang bc0e271e71 [buider] use builder() for cpu adam and fused optim in setup.py (#2187 )

2022-12-23 16:05:13 +08:00

3.5 KiB

Raw Blame History

View Raw