ColossalAI/colossalai/testing
jiangmingyan 20068ba188
[booster] add tests for ddp and low level zero's checkpointio (#3715)
* [booster] update tests for booster

* [booster] update tests for booster

* [booster] update tests for booster

* [booster] update tests for booster

* [booster] update tests for booster

* [booster] update booster tutorials#3717, fix recursive check
2023-05-10 12:17:02 +08:00
..
__init__.py [booster] add tests for ddp and low level zero's checkpointio (#3715) 2023-05-10 12:17:02 +08:00
comparison.py [booster] add tests for ddp and low level zero's checkpointio (#3715) 2023-05-10 12:17:02 +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 [doc] Fix typo under colossalai and doc(#3618) 2023-04-26 11:38:43 +08:00