You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/utils/checkpoint
HELSON 8463290642
[checkpoint] use args, kwargs in save_checkpoint, load_checkpoint (#1368)
2 years ago
..
__init__.py [checkpoint] checkpoint for ColoTensor Model (#1196) 2 years ago
module_checkpoint.py [checkpoint] use args, kwargs in save_checkpoint, load_checkpoint (#1368) 2 years ago
utils.py [colotensor] use cpu memory to store state_dict (#1367) 2 years ago