ytxiong a017cab4b3 fix(*): move sequence_parallel to parallel config (#224) ... * move sequence_parallel to parallel config * set the sequece_parallel default value is False * fix lint * fix lint * fix lint		2023-08-24 09:49: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(train.py): support torch profiler (#201)	2023-08-21 15:23:38 +08:00
evaluation.py	fix(*): move sequence_parallel to parallel config (#224)	2023-08-24 09:49:04 +08:00
logger.py	feat(utils/evaluation.py): support evaluate (#154)	2023-08-02 19:03:59 +08:00
megatron_timers.py	Feat/overlap_bcast_forward (#218)	2023-08-23 16:59:59 +08:00
model_checkpoint.py	Feat/overlap_bcast_forward (#218)	2023-08-23 16:59:59 +08:00
parallel.py	feat(utils/logger.py): support uniscale logger (#152)	2023-08-01 17:37:32 +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	feat(ckpt): add auto ckpt load and singal quit (#216)	2023-08-23 14:17:45 +08:00
timeout.py	Merge main to develop (#203)	2023-08-16 15:57:26 +08:00
writer.py	feat(ckpt): add auto ckpt load and singal quit (#216)	2023-08-23 14:17:45 +08:00