ColossalAI/colossalai/testing
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
..
__init__.py
comparison.py [amp] add gradient clipping for unit tests (#2283) 2023-01-04 11:59:56 +08:00
pytest_wrapper.py
random.py
utils.py