zhanglei ccdaf8ec45 fix the moe_loss for ci and val		2023-09-22 15:45:36 +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	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
evaluation.py	fix the moe_loss for ci and val	2023-09-22 15:45:36 +08:00
gputest.py	Feat(PythonGC): Do garbage collection manually (#326)	2023-09-22 13:52:25 +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 internlm/develop into feature_add_moe	2023-09-19 13:27:43 +08:00
parallel.py	refactor code	2023-09-19 12:30:40 +08:00
registry.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
simple_memory_profiler.py	Merge develop to main (#233)	2023-08-24 22:03:04 +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	feat(utils/writer.py): support writer add_scalars for writing dict data (#257)	2023-09-01 13:24:46 +08:00