mirror of https://github.com/InternLM/InternLM
* change mode to dummy in model construction and restore to data when done * add comments * move set_mode(.DATA) to initialize_model(.) |
||
|---|---|---|
| .. | ||
| communication | ||
| context | ||
| scheduler | ||
| __init__.py | ||
| engine.py | ||
| gradient_handler.py | ||
| naive_amp.py | ||
| trainer.py | ||