ColossalAI/colossalai/utils
Frank Lee 9a0466534c
update markdown docs (english) (#60)
2021-12-10 14:37:33 +08:00
..
data_sampler update markdown docs (english) (#60) 2021-12-10 14:37:33 +08:00
gradient_accumulation Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
multi_tensor_apply Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
__init__.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
activation_checkpoint.py Migrated project 2021-10-28 18:21:23 +02:00
checkpointing.py Support TP-compatible Torch AMP and Update trainer API (#27) 2021-11-18 19:45:06 +08:00
common.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
cuda.py Migrated project 2021-10-28 18:21:23 +02:00
memory.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
timer.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00