ColossalAI/tests/test_booster/test_plugin
Frank Lee 638a07a7f9
[test] fixed gemini plugin test (#3411)
* [test] fixed gemini plugin test

* polish code

* polish code
2023-04-03 17:12:22 +08:00
..
test_gemini_plugin.py [test] fixed gemini plugin test (#3411) 2023-04-03 17:12:22 +08:00
test_torch_ddp_plugin.py [booster] implemented the torch ddd + resnet example (#3232) 2023-03-27 10:24:14 +08:00