InternLM/internlm/initialize
mwiacx 64f0d838e2 update args sanity check for cuda memory balance 2023-09-22 10:10:58 +08:00
..
legacy feat(ckpt): fix checkpoint bugs and add feature enhancements. (#259) 2023-09-05 17:40:48 +08:00
__init__.py Merge develop to main (#233) 2023-08-24 22:03:04 +08:00
initialize_tensor.py feat(model): implement uniform_init for tensor. (#252) 2023-09-01 01:12:53 +08:00
initialize_trainer.py docs(*): add documentation and reST files for readthedocs (#272) 2023-09-06 15:36:03 +08:00
launch.py update args sanity check for cuda memory balance 2023-09-22 10:10:58 +08:00