ColossalAI/tests/test_tensor/model
HELSON 7a065dc9f6
[hotfix] fix megatron_init in test_gpt2.py (#1357)
2022-07-25 10:28:19 +08:00
..
test_gpt2.py [hotfix] fix megatron_init in test_gpt2.py (#1357) 2022-07-25 10:28:19 +08:00
test_model.py [refactor] refactor ColoTensor's unit tests (#1340) 2022-07-19 15:46:24 +08:00
test_module_spec.py [refactor] refactor ColoTensor's unit tests (#1340) 2022-07-19 15:46:24 +08:00