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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-25 09:12:10 +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