mirror of https://github.com/hpcaitech/ColossalAI
077a5cdde4
* [zero] fix gradient clipping in hybrid parallelism * [testing] change model name to avoid pytest warning * [hotfix] fix unit testing |
||
|---|---|---|
| .. | ||
| grad_scaler | ||
| __init__.py | ||
| _fp16_optimizer.py | ||
| _utils.py | ||
| naive_amp.py | ||