• Olatunji Ruwase's avatar
    Support multi-output models (#170) · 53c73fe3
    Olatunji Ruwase authored
    * 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
    53c73fe3
multi_output_model.py 1.41 KB