Fix EMNIST action when MPS is available

Tony-Y · Oct 23, 2024 · dc24d94 · dc24d94
1 parent c7e1740
commit dc24d94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/example-emnist.yml b/.github/workflows/example-emnist.yml
@@ -48,4 +48,4 @@ jobs:
     - name: Download EMNIST dataset
       run: python examples/emnist/download.py
     - name: Train a model on EMNIST dataset
-      run: python examples/emnist/main.py --epochs 1
+      run: python examples/emnist/main.py --epochs 1 --no-gpu