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