Patch GPU failures (#6281)

* Pin to 1.5.0 * Patch XLM GPU test
2020-08-07 02:58:15 -04:00
parent 80a0676a51
commit 0d9328f2ef
2 changed files with 4 additions and 2 deletions
--- a/.github/workflows/self-push.yml
+++ b/.github/workflows/self-push.yml
@@ -36,7 +36,7 @@ jobs:
      run: |
        source .env/bin/activate
        pip install --upgrade pip
-        pip install torch --no-cache-dir
+        pip install torch!=1.6.0 --no-cache-dir
        pip install .[sklearn,testing]

    - name: Are GPUs recognized by our DL frameworks