InternLM/internlm/core
Wenwen Qu 7f687bf4b3
fix(core/context): use dummy mode to generate random numbers in model construction (#266)
* change mode to dummy in model construction and restore to data when done

* add comments

* move set_mode(.DATA) to initialize_model(.)
2023-09-06 14:34:11 +08:00
..
communication Merge develop to main (#233) 2023-08-24 22:03:04 +08:00
context fix(core/context): use dummy mode to generate random numbers in model construction (#266) 2023-09-06 14:34:11 +08:00
scheduler Merge develop to main (#233) 2023-08-24 22:03:04 +08:00
__init__.py initial commit 2023-07-06 12:55:23 +08:00
engine.py initial commit 2023-07-06 12:55:23 +08:00
gradient_handler.py initial commit 2023-07-06 12:55:23 +08:00
naive_amp.py Merge develop to main (#233) 2023-08-24 22:03:04 +08:00
trainer.py feat(ckpt): fix checkpoint bugs and add feature enhancements. (#259) 2023-09-05 17:40:48 +08:00