Support multi-output models (#170)
* Push to remote * Correctly handle multi output models by doing loss scaling in backward() Unit tests for multi output models * Fix formatting issues * Formatting issues fix * Fix formatting * Update DeepSpeedExamples submodule Enable Megatron model tests
Showing
tests/unit/multi_output_model.py
0 → 100755
想要评论请 注册 或 登录