HELSON 87775a0682 [colotensor] use cpu memory to store state_dict (#1367)

2022-07-26 14:13:38 +08:00

..

__init__.py

[checkpoint] checkpoint for ColoTensor Model (#1196)

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

module_checkpoint.py

[hotfix] fix no optimizer in save/load (#1363)

2022-07-26 10:53:53 +08:00

utils.py

[colotensor] use cpu memory to store state_dict (#1367)

2022-07-26 14:13:38 +08:00