Wenwen Qu 8a595837fc merge upstream/develop into feature_add_moe		2023-09-11 16:20:08 +08:00
..
__init__.py	initial commit	2023-07-06 12:55:23 +08:00
checkpoint.py	initial commit	2023-07-06 12:55:23 +08:00
common.py	feat(train.py): support torch profiler (#201)	2023-08-21 15:23:38 +08:00
evaluation.py	merge upstream/develop into feature_add_moe	2023-09-11 16:20:08 +08:00
gputest.py	merge upstream/develop into feature_add_moe	2023-09-11 16:20:08 +08:00
logger.py	merge upstream/develop into feature_add_moe	2023-09-11 16:20:08 +08:00
megatron_timers.py	merge upstream/develop into feature_add_moe	2023-09-11 16:20:08 +08:00
model_checkpoint.py	merge upstream/develop into feature_add_moe	2023-09-11 16:20:08 +08:00
parallel.py	fix moe bugs in zero optimizer	2023-08-17 16:11:34 +08:00
registry.py	feat(core/scheduler): support pipeline parallel (#98)	2023-07-24 20:52:09 +08:00
simple_memory_profiler.py	feat(memory_profiler): improve memory profiler (#217)	2023-08-23 14:18:33 +08:00
storage_manager.py	merge upstream/develop into feature_add_moe	2023-09-11 16:20:08 +08:00
timeout.py	merge upstream/develop into feature_add_moe	2023-09-11 16:20:08 +08:00
writer.py	remove moe_loss_coeff parameter passing	2023-08-31 18:44:58 +08:00