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_torch_fp16.py [amp] add torch amp test (#1860) 2022-11-10 16:40:26 +08:00