[upd] typo

2 years ago · 46f9ad2871
parent 5fbd93d3df
commit 46f9ad2871
1 changed files with 2 additions and 2 deletions
--- a/pretrain/viz_reconstruction.ipynb
+++ b/pretrain/viz_reconstruction.ipynb
@ -41,8 +41,8 @@
    "\n",
    "# specify the device to use\n",
    "USING_GPU_IF_AVAILABLE = True\n",
-    "# specify the CNN to useTrue for ConvNeXt-L-384, False for ConvNeXt-S-224\n",
-    "USING_LARGE384_MODEL = True    #\n",
+    "# specify the CNN to use\n",
+    "USING_LARGE384_MODEL = True    # True for ConvNeXt-L-384, False for ConvNeXt-S-224\n",
    "import torch\n",
    "_ = torch.empty(1)\n",
    "if torch.cuda.is_available() and USING_GPU_IF_AVAILABLE:\n",