ColossalAI/tests/test_zero/test_state_dict.py at e99edfcb51df48dec17498c60bbbd06baa293c22 - 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

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