lijiaxing 7d67795f7e add assert		2023-12-14 15:01: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	fix(pp): fix no-packed dataset load micro batch error (#538)	2023-12-13 14:48:32 +08:00
evaluation.py	1. fix(config): rampup_batch_size defalut value BC. (#515)	2023-11-28 19:33:46 +08:00
gputest.py	1. fix(config): rampup_batch_size defalut value BC. (#515)	2023-11-28 19:33:46 +08:00
logger.py	feat(utils): add timeout warpper for key functions (#286)	2023-09-07 17:26:17 +08:00
megatron_timers.py	feat: add runtime diag (#297)	2023-09-08 17:56:46 +08:00
model_checkpoint.py	add assert	2023-12-14 15:01:04 +08:00
parallel.py	fix(model): add IS_SEQUENCE_PARALLEL check for norm module (#528)	2023-12-06 12:06:22 +08:00
registry.py	fix bug	2023-12-13 15:14:36 +08:00
simple_memory_profiler.py	fix(moe): fix moe compatibility for fsdp and memory profiling (#417)	2023-10-17 14:13:48 +08:00
storage_manager.py	feat(ckpt): support auto resume in Volc and Ali (#529)	2023-12-12 13:27:24 +08:00
timeout.py	fix(timeout): larger timeout (#495)	2023-11-21 19:19:22 +08:00
writer.py	fix(train): unify the exp paths (#492)	2023-11-11 20:15:59 +08:00