Modify autoencoderkl as per #1924.

Signed-off-by: Virginia Fernandez <[email protected]>
Project-MONAI · Jan 20, 2025 · d24a090 · d24a090
1 parent 3617388
commit d24a090
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/generation/3d_autoencoderkl/3d_autoencoderkl_tutorial.ipynb b/generation/3d_autoencoderkl/3d_autoencoderkl_tutorial.ipynb
@@ -178,6 +178,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "channel = 0  # 0 = Flair\n",
+    "assert channel in [0, 1, 2, 3], \"Choose a valid channel\"\n",
     "all_transforms = [\n",
     "    transforms.LoadImaged(keys=[\"image\"]),\n",
     "    transforms.EnsureChannelFirstd(keys=[\"image\"]),\n",
@@ -207,8 +209,6 @@
     }
    ],
    "source": [
-    "channel = 0  # 0 = Flair\n",
-    "assert channel in [0, 1, 2, 3], \"Choose a valid channel\"\n",
     "# Training\n",
     "train_ds = DecathlonDataset(\n",
     "    root_dir=root_dir,\n",
@@ -785,7 +785,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.0"
+   "version": "3.10.15"
   }
  },
  "nbformat": 4,