onnxruntime/tools/ci_build/github/linux/docker/scripts/training
baijumeswani 249a2c14ef
Pin version of pytorch to 1.8.1 for ORTModule CI pipeline (#7167)
* Pin version of pytorch to 1.8.1 for ORTModule CI pipeline
* Use pytorch-lightning stable version 1.2.5
* Revert to cuda 10.1
2021-04-01 09:37:47 -07:00
..
ortmodule
azure_scale_set_vm_mount_test_data.sh
requirements.txt