ColossalAI/colossalai/testing
HELSON a1ce02d740
[zero] test gradient accumulation (#1964)
* [zero] fix memory leak for zero2

* [zero] test gradient accumulation

* [zero] remove grad clip test
2022-11-29 13:00:30 +08:00
..
__init__.py
comparison.py [NFC] polish colossalai/testing/comparison.py code style. (#1558) 2022-09-08 22:11:04 +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