fix notebooks

Author: LucaMarconato

notebooks/examples/densenet.ipynb

@@ -1109,7 +1109,7 @@
     "\n",
     "PATH_DATASETS = os.environ.get(\"PATH_DATASETS\", \"..\")\n",
     "BATCH_SIZE = 4096 if torch.cuda.is_available() else 64\n",
-    "NUM_WORKERS = 10 if torch.cuda.is_available() else 8\n",
+    "NUM_WORKERS = 10 if torch.cuda.is_available() else 4\n",
     "print(f\"Using {BATCH_SIZE} batch size.\")\n",
     "print(f\"Using {NUM_WORKERS} workers.\")\n",
     "\n",
@@ -1628,6 +1628,8 @@
     }
    ],
    "source": [
+    "import matplotlib.pyplot as plt\n",
+    "\n",
     "axes = plt.subplots(1, 2, figsize=(15, 7))[1]\n",
     "(\n",
     "    merged.pl.render_images(\"CytAssist_FFPE_Human_Breast_Cancer_full_image\")\n",