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 Pin version of pytorch to 1.8.1 for ORTModule CI pipeline (#7167) 2021-04-01 09:37:47 -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