Fix

Signed-off-by: Eric Kerfoot <[email protected]>
Project-MONAI · Sep 6, 2024 · 6ec6692 · 6ec6692
1 parent f384232
commit 6ec6692
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git a/generation/2d_vqgan/2d_vqgan_tutorial.ipynb b/generation/2d_vqgan/2d_vqgan_tutorial.ipynb
@@ -222,6 +222,8 @@
     }
    ],
    "source": [
+    "batch_size = 16\n",
+    "\n",
     "train_transforms = mt.Compose(\n",
     "    [\n",
     "        mt.LoadImaged(keys=[\"image\"]),\n",
@@ -239,7 +241,7 @@
     "    ]\n",
     ")\n",
     "train_ds = CacheDataset(data=train_datalist, transform=train_transforms)\n",
-    "train_loader = DataLoader(train_ds, batch_size=256, shuffle=True, num_workers=4, persistent_workers=True)"
+    "train_loader = DataLoader(train_ds, batch_size=batch_size, shuffle=True, num_workers=4, persistent_workers=True)"
    ]
   },
   {
@@ -276,7 +278,7 @@
     "    ]\n",
     ")\n",
     "val_ds = CacheDataset(data=val_datalist, transform=val_transforms)\n",
-    "val_loader = DataLoader(val_ds, batch_size=256, shuffle=False, num_workers=4, persistent_workers=True)"
+    "val_loader = DataLoader(val_ds, batch_size=batch_size, shuffle=False, num_workers=4, persistent_workers=True)"
    ]
   },
   {

diff --git a/generation/2d_vqvae/2d_vqvae_tutorial.ipynb b/generation/2d_vqvae/2d_vqvae_tutorial.ipynb
@@ -221,6 +221,8 @@
     "train_data = MedNISTDataset(root_dir=root_dir, section=\"training\", download=True, seed=0)\n",
     "train_datalist = [{\"image\": item[\"image\"]} for item in train_data.data if item[\"class_name\"] == \"HeadCT\"]\n",
     "image_size = 64\n",
+    "batch_size = 16\n",
+    "\n",
     "train_transforms = mt.Compose(\n",
     "    [\n",
     "        mt.LoadImaged(keys=[\"image\"]),\n",
@@ -238,7 +240,7 @@
     "    ]\n",
     ")\n",
     "train_ds = Dataset(data=train_datalist, transform=train_transforms)\n",
-    "train_loader = DataLoader(train_ds, batch_size=64, shuffle=True, num_workers=4, persistent_workers=True)"
+    "train_loader = DataLoader(train_ds, batch_size=batch_size, shuffle=True, num_workers=4, persistent_workers=True)"
    ]
   },
   {
@@ -317,7 +319,7 @@
     "    ]\n",
     ")\n",
     "val_ds = Dataset(data=val_datalist, transform=val_transforms)\n",
-    "val_loader = DataLoader(val_ds, batch_size=64, shuffle=True, num_workers=4, persistent_workers=True)"
+    "val_loader = DataLoader(val_ds, batch_size=batch_size, shuffle=True, num_workers=4, persistent_workers=True)"
    ]
   },
   {

diff --git a/generation/2d_vqvae_transformer/2d_vqvae_transformer_tutorial.ipynb b/generation/2d_vqvae_transformer/2d_vqvae_transformer_tutorial.ipynb
@@ -226,6 +226,7 @@
     "train_data = MedNISTDataset(root_dir=root_dir, section=\"training\", download=True, seed=0)\n",
     "train_datalist = [{\"image\": item[\"image\"]} for item in train_data.data if item[\"class_name\"] == \"HeadCT\"]\n",
     "image_size = 64\n",
+    "batch_size = 16\n",
     "train_transforms = transforms.Compose(\n",
     "    [\n",
     "        transforms.LoadImaged(keys=[\"image\"]),\n",
@@ -243,7 +244,7 @@
     "    ]\n",
     ")\n",
     "train_ds = Dataset(data=train_datalist, transform=train_transforms)\n",
-    "train_loader = DataLoader(train_ds, batch_size=128, shuffle=True, num_workers=4, persistent_workers=True)"
+    "train_loader = DataLoader(train_ds, batch_size=batch_size, shuffle=True, num_workers=4, persistent_workers=True)"
    ]
   },
   {
@@ -322,7 +323,7 @@
     "    ]\n",
     ")\n",
     "val_ds = Dataset(data=val_datalist, transform=val_transforms)\n",
-    "val_loader = DataLoader(val_ds, batch_size=128, shuffle=True, num_workers=4, persistent_workers=True)"
+    "val_loader = DataLoader(val_ds, batch_size=batch_size, shuffle=True, num_workers=4, persistent_workers=True)"
    ]
   },
   {