diff --git a/pretrain/viz_reconstruction.ipynb b/pretrain/viz_reconstruction.ipynb
index 9c76dcf..19fda7f 100644
--- 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",