ColossalAI/tests/test_amp at cc0ed7cf33f379d1954bed9360d528df1bab1ff6 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-12 07:02:22 +00:00

Files

History

xcnick a141681260 [amp] add torch amp test (#1860 )

2022-11-10 16:40:26 +08:00

..

test_naive_fp16.py

[test] refactored with the new rerun decorator (#763 )

2022-04-15 00:33:04 +08:00

test_torch_fp16.py

[amp] add torch amp test (#1860 )

2022-11-10 16:40:26 +08:00