Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jan 9, 2024
1 parent aecbd76 commit 5b06c1e
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions modules/torch_compile.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -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",
")"
]
Expand Down Expand Up @@ -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)"
]
Expand Down

0 comments on commit 5b06c1e

Please sign in to comment.