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