chore: remove last MPS occurence

zama-ai · Jan 8, 2024 · e2b083c · e2b083c
1 parent ccc484b
commit e2b083c
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/docs/advanced_examples/QuantizationAwareTraining.ipynb b/docs/advanced_examples/QuantizationAwareTraining.ipynb
@@ -400,11 +400,10 @@
     }
    ],
    "source": [
-    "# You can also train this on GPU for example by setting\n",
-    "# MPS option is supported by macOS with Apple Silicon or AMD GPUs\n",
-    "device = (\n",
-    "    \"cuda\" if torch.cuda.is_available() else \"mps\" if torch.backends.mps.is_available() else \"cpu\"\n",
-    ")\n",
+    "# Add MPS (for macOS with Apple Silicon or AMD GPUs) support when error is fixed. For now, we\n",
+    "# observe a decrease in torch's top1 accuracy when using MPS devices\n",
+    "# FIXME: https://github.com/zama-ai/concrete-ml-internal/issues/3953\n",
+    "device = \"cuda\" if torch.cuda.is_available() else \"cpu\"\n",
     "\n",
     "# Define our loss function\n",
     "criterion = nn.CrossEntropyLoss()\n",