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

[test] skip tests when not enough GPUs are detected (#1090)

2022-06-09 17:19:13 +08:00

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