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
HELSON 7a065dc9f6
[hotfix] fix megatron_init in test_gpt2.py (#1357)
2 years ago
..
test_gpt2.py [hotfix] fix megatron_init in test_gpt2.py (#1357) 2 years ago
test_model.py [refactor] refactor ColoTensor's unit tests (#1340) 2 years ago
test_module_spec.py [refactor] refactor ColoTensor's unit tests (#1340) 2 years ago