azureml-examples
azureml-examples copied to clipboard
Use PyTorchConfiguration (instead of MpiConfiguration) for DeepSpeed
PR into Azure/azureml-examples
Checklist
I have:
- [X] read and followed the contributing guidelines
Changes
- MpiConfiguration was replaced with PyTorchConfiguration in the DeepSpeed example with CIFAR dataset