Spaces:

zaidmehdi
/

arabic-dialect-classifier

Sleeping

zaidmehdi commited on Feb 26

Commit

40d1a73

•

1 Parent(s): cf5faeb

changing torch device to cuda

Files changed (1) hide show

src/classifier.ipynb CHANGED Viewed

@@ -393,7 +393,7 @@
     }
    ],
    "source": [
-    "device = torch.device(\"cpu\")\n",
     "type(device)"
    ]
   },

     }
    ],
    "source": [
+    "device = torch.device(\"cuda\")\n",
     "type(device)"
    ]
   },