ColossalAI/tests/test_zero/test_state_dict.py at e94c79f15b2ee1428ccda2cdb2d00007697c578c - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-14 22:28:18 +00:00

Files

ver217 7a05367101 [hotfix] shared model returns cpu state_dict (#1328 )

2022-07-15 22:11:37 +08:00

2.2 KiB

Raw Blame History

View Raw