Commits
baijumeswani authored and GitHub committed 249a2c14eff
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