optimum
021ae43f - Patch ORTTrainer's compatibility with DeepSpeed (#148)

Commit
4 years ago
Patch ORTTrainer's compatibility with DeepSpeed (#148) * Fix missing dependencies * Unwrap model to get ORTModule & add ds configs for further tests * Validate compatibility with fairscale * Fix compatibility of ort inference with DeepSpeed * Modify the _export with the flag in transformers.onnx.export * Add transformers version demand * Add assets for testing BF16
Author
Parents
Loading