HELSON 8463290642 [checkpoint] use args, kwargs in save_checkpoint, load_checkpoint (#1368)

2022-07-26 14:41:53 +08:00

..

__init__.py

[checkpoint] checkpoint for ColoTensor Model (#1196)

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

module_checkpoint.py

[checkpoint] use args, kwargs in save_checkpoint, load_checkpoint (#1368)

2022-07-26 14:41:53 +08:00

utils.py

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

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