ColossalAI/tests/test_amp at 28e55c2530cc65859129a893affef63ecb6b3549 - 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