onnxruntime/tools/ci_build/github/linux/docker/scripts/training
harshithapv 540eac253e
Deepspeed pipeline parallel and fairscale sharded optimizer test samples with ORTModule (#7078)
* adding samples for Deepspeed pipeline parallel and fairscale sharded optimizer with ortmodule

* fixed typo in args

* addressed Thiago's comments

* Update orttraining/orttraining/test/python/orttraining_test_ortmodule_deepspeed_pipeline_parallel.py

Co-authored-by: Thiago Crepaldi <thiago.crepaldi@microsoft.com>
2021-03-24 09:43:05 -07:00
..
ortmodule Deepspeed pipeline parallel and fairscale sharded optimizer test samples with ORTModule (#7078) 2021-03-24 09:43:05 -07:00
azure_scale_set_vm_mount_test_data.sh Cache datasets on CI machines (#6525) 2021-02-02 21:11:35 -08:00
requirements.txt Separate requirements.txt file for ORTModule pipelines (#6879) 2021-03-05 14:12:11 -08:00