zhanglei 7b1709a7ff Merge branch 'feature_add_moe' of github.com:blankde/InternLM into feature_add_moe_pp_zl ... Conflicts: train.py		2023-08-17 17:00:04 +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(monitor): support monitor and alert (#175)	2023-08-08 11:18:15 +08:00
evaluation.py	add logger for moe_loss	2023-08-17 16:52:11 +08:00
logger.py	feat(utils/evaluation.py): support evaluate (#154)	2023-08-02 19:03:59 +08:00
megatron_timers.py	initial commit	2023-07-06 12:55:23 +08:00
model_checkpoint.py	Revert "feat(ckpt): add auto ckpt load and singal quit (#189)" (#192)	2023-08-11 17:12:26 +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	Merge main to develop (#203)	2023-08-16 15:57:26 +08:00
storage_manager.py	fix/ci train error (#200)	2023-08-16 11:11:27 +08:00
timeout.py	Merge main to develop (#203)	2023-08-16 15:57:26 +08:00
writer.py	feat(utils/logger.py): support uniscale logger (#152)	2023-08-01 17:37:32 +08:00