ColossalAI/colossalai/utils/checkpoint at 942c8cd1fb509d32fc741d076b6ed0d80b03902b - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-17 16:08:30 +00:00

Files

History

HELSON f92c100ddd [checkpoint] use gather_tensor in checkpoint and update its unit test (#1339 )

2022-07-19 14:15:28 +08:00

..

__init__.py

[checkpoint] checkpoint for ColoTensor Model (#1196 )

2022-07-06 17:22:03 +08:00

module_checkpoint.py

[checkpoint] use gather_tensor in checkpoint and update its unit test (#1339 )

2022-07-19 14:15:28 +08:00

utils.py

[checkpoint] use gather_tensor in checkpoint and update its unit test (#1339 )

2022-07-19 14:15:28 +08:00