Fix

Signed-off-by: Eric Kerfoot <[email protected]>
Project-MONAI · Sep 5, 2024 · 76db892 · 76db892
1 parent 3a9cd2d
commit 76db892
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/generation/2d_ddpm/2d_ddpm_inpainting.ipynb b/generation/2d_ddpm/2d_ddpm_inpainting.ipynb
@@ -212,7 +212,7 @@
    "source": [
     "train_data = MedNISTDataset(root_dir=root_dir, section=\"training\", download=True, progress=False, seed=0)\n",
     "train_datalist = [{\"image\": item[\"image\"]} for item in train_data.data if item[\"class_name\"] == \"Hand\"]\n",
-    "batch_size = 128\n",
+    "batch_size = 64\n",
     "num_workers = 4"
    ]
   },