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
Frank Lee 9a0466534c
update markdown docs (english) (#60)
3 years ago
..
data_sampler update markdown docs (english) (#60) 3 years ago
gradient_accumulation Develop/experiments (#59) 3 years ago
multi_tensor_apply Develop/experiments (#59) 3 years ago
__init__.py Develop/experiments (#59) 3 years ago
activation_checkpoint.py Migrated project 3 years ago
checkpointing.py Support TP-compatible Torch AMP and Update trainer API (#27) 3 years ago
common.py Develop/experiments (#59) 3 years ago
cuda.py Migrated project 3 years ago
memory.py Develop/experiments (#59) 3 years ago
timer.py Develop/experiments (#59) 3 years ago