ColossalAI/tests/test_tensor/model at c491c2a948af77d07b4a35fd09a985d5b65a9087 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-03 17:37:23 +00:00

Files

History

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