github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-27 02:03:13 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
d42aecdda1f81f3777bc4c5788df39ef2ec2e04f
ColossalAI/tests/test_amp
History
HELSON 5d3a2be3af [amp] add gradient clipping for unit tests (#2283) 
* [amp] add gradient clipping in unit tests

* fix bugs
2023-01-04 11:59:56 +08:00
..
test_naive_fp16.py
[amp] add gradient clipping for unit tests (#2283)
2023-01-04 11:59:56 +08:00
test_torch_fp16.py
[amp] add gradient clipping for unit tests (#2283)
2023-01-04 11:59:56 +08:00