ee414d25be
* switch to 1.6 * readme * 1.7 * back to normal [ci skip] * horovodrun --verbose * try with apex * add apex test * change base * description * test with 1.7 * back to 1.6 * no gradient_clip_val * re-add gradient_clip_val * no amp * temp skip torch.cuda.amp + horovod test * Apply suggestion from code review Co-authored-by: Sean Naren <sean.narenthiran@gmail.com> * Fix formatting * ddp * Moved extended model outside of function to prevent pickling issue for drone * typo * resolve bug * extract automatic_automization Co-authored-by: Sean Naren <sean.narenthiran@gmail.com> Co-authored-by: SeanNaren <sean@grid.ai> Co-authored-by: chaton <thomas@grid.ai> |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| test_amp_plugin.py | ||
| test_apex_plugin.py | ||
| test_ddp_plugin.py | ||