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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-05 07:34:30 +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