tests/test_zero_data_parallel/test_sharded_optim.py · 526a3180328b41a51c9a5ede0e71d2017d61736d · OpenDAS / ColossalAI · GitLab

Find file Blame History Permalink

[unit test] Refactored test cases with component func (#339) · 526a3180
Frank Lee authored Mar 11, 2022
```
* refactored test with component func

* fixed bug
```
526a3180

test_sharded_optim.py 5.61 KB