diff --git a/examples/quickstart.ipynb b/examples/quickstart.ipynb index 2abaa53b..262004a1 100644 --- a/examples/quickstart.ipynb +++ b/examples/quickstart.ipynb @@ -88,8 +88,8 @@ "outputs": [], "source": [ "batch_size = 128\n", - "data_path='/tmp/data/mnist'\n", - "device = torch.device(\"cuda\") if torch.cuda.is_available() else torch.device("mps") if torch.backends.mps.is_available() else torch.device(\"cpu\")" + "data_path='/data/mnist'\n", + "device = torch.device(\"cuda\") if torch.cuda.is_available() else torch.device(\"cpu\")" ] }, {