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 [unittest] added doc for the pytest wrapper (#1704) 2022-10-14 10:56:17 +08:00
random.py [zero] test gradient accumulation (#1964) 2022-11-29 13:00:30 +08:00
utils.py [unittest] supported condititonal testing based on env var (#1701) 2022-10-13 19:38:45 +08:00