Jiarui Fang 20da6e48c8 [checkpoint] save sharded optimizer states (#1237)

2022-07-08 16:33:13 +08:00

..

__init__.py

[checkpoint] checkpoint for ColoTensor Model (#1196)

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

module_checkpoint.py

[checkpoint] save sharded optimizer states (#1237)

2022-07-08 16:33:13 +08:00