add cpu-only mode

notebooks/demo.ipynb

@@ -65,26 +65,10 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 2,
    "id": "1745a859-71d4-4ec3-8ef3-027cabe786d4",
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "/private/home/nikitakaraev/dev/neurips_2023/co-tracker\n"
-     ]
-    },
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "/private/home/nikitakaraev/.conda/envs/stereoformer/lib/python3.8/site-packages/requests/__init__.py:109: RequestsDependencyWarning: urllib3 (1.26.14) or chardet (None)/charset_normalizer (3.2.0) doesn't match a supported version!\n",
-      "  warnings.warn(\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "%cd ..\n",
     "import os\n",
@@ -105,7 +89,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 3,
    "id": "f1f9ca4d-951e-49d2-8844-91f7bcadfecd",
    "metadata": {},
    "outputs": [],
@@ -116,7 +100,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 4,
    "id": "fb4c2e9d-0e85-4c10-81a2-827d0759bf87",
    "metadata": {},
    "outputs": [
@@ -129,7 +113,7 @@
        "<IPython.core.display.HTML object>"
       ]
      },
-     "execution_count": 3,
+     "execution_count": 4,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -175,8 +159,8 @@
    "outputs": [],
    "source": [
     "if torch.cuda.is_available():\n",
-    "    model=model.cuda()\n",
-    "    video=video.cuda()"
+    "    model = model.cuda()\n",
+    "    video = video.cuda()"
    ]
   },
   {
@@ -282,7 +266,9 @@
     "    [10., 600., 500.], # frame number 10\n",
     "    [20., 750., 600.], # ...\n",
     "    [30., 900., 200.]\n",
-    "]).cuda()"
+    "])\n",
+    "if torch.cuda.is_available():\n",
+    "    queries = queries.cuda()"
    ]
   },
   {