You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_tensor/model
Jiarui Fang 3ce4463fe6
[utils] remove lazy_memory_allocate from ColoInitContext (#1844)
2 years ago
..
test_gpt2.py [hotfix] fix megatron_init in test_gpt2.py (#1357) 2 years ago
test_model.py [utils] remove lazy_memory_allocate from ColoInitContext (#1844) 2 years ago
test_module_spec.py [utils] remove lazy_memory_allocate from ColoInitContext (#1844) 2 years ago