diff --git a/modules/torch_compile.ipynb b/modules/torch_compile.ipynb index 9065aab720..440add1d26 100644 --- a/modules/torch_compile.ipynb +++ b/modules/torch_compile.ipynb @@ -195,7 +195,7 @@ " mt.RandZoomD(keys=(\"image\", \"label\"), prob=1.0, min_zoom=0.8, max_zoom=1.2, mode=(\"trilinear\", \"nearest\")),\n", " mt.ResizeWithPadOrCropD(keys=(\"image\", \"label\"), spatial_size=(96, 96, 96)),\n", " # add `FromMetaTensorD` to convert `MetaTensor` to `torch.Tensor`\n", - " mt.FromMetaTensorD(keys=(\"image\", \"label\"))\n", + " mt.FromMetaTensorD(keys=(\"image\", \"label\")),\n", " ]\n", ")" ] @@ -245,9 +245,10 @@ "os.makedirs(bundle_dir, exist_ok=True)\n", "\n", "bundle = download(\"wholeBody_ct_segmentation\", bundle_dir=bundle_dir)\n", - "config_file = os.path.join(bundle_dir, 'wholeBody_ct_segmentation/configs/train.json')\n", + "config_file = os.path.join(bundle_dir, \"wholeBody_ct_segmentation/configs/train.json\")\n", "train_workflow = create_workflow(config_file=str(config_file), workflow_type=\"train\")\n", "\n", + "\n", "def create_model():\n", " return train_workflow.network_def.to(device)" ]