InternLM/internlm/utils
Sun Peng 1ee31ff9b1
feat: add runtime diag (#297)
* feat: add runtime diag

* add diag_outlier_ratio

---------

Co-authored-by: yingtongxiong <974106207@qq.com>
2023-09-08 17:56:46 +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(eval): no need to check length of valid_dl when using streaming dataset (#274) 2023-09-04 23:14:07 +08:00
gputest.py feat: add runtime diag (#297) 2023-09-08 17:56: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 feat(utils): add timeout warpper for key functions (#286) 2023-09-07 17:26:17 +08:00
parallel.py Merge develop to main (#233) 2023-08-24 22:03:04 +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 fix(storage): fix and refactor storage api (#281) 2023-09-06 01:15:09 +08:00
timeout.py feat(utils): add timeout warpper for key functions (#286) 2023-09-07 17:26:17 +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