diff --git a/bundle/pythonic_usage_guidance/pythonic_bundle_access.ipynb b/bundle/pythonic_usage_guidance/pythonic_bundle_access.ipynb index 2360602142..84535100c7 100644 --- a/bundle/pythonic_usage_guidance/pythonic_bundle_access.ipynb +++ b/bundle/pythonic_usage_guidance/pythonic_bundle_access.ipynb @@ -404,8 +404,7 @@ " name=\"brats_mri_segmentation\",\n", " bundle_dir=root_dir,\n", " source=\"monaihosting\",\n", - " in_channels=3,\n", - " out_channels=1,\n", + " net_override={\"in_channels\": 3, \"out_channels\": 1},\n", ")\n", "\n", "# using `exclude_vars` to filter loading weights.\n", diff --git a/computer_assisted_intervention/endoscopic_inbody_classification.ipynb b/computer_assisted_intervention/endoscopic_inbody_classification.ipynb index 6e095f3019..fb97c7f474 100644 --- a/computer_assisted_intervention/endoscopic_inbody_classification.ipynb +++ b/computer_assisted_intervention/endoscopic_inbody_classification.ipynb @@ -36,7 +36,6 @@ "execution_count": 1, "id": "5f9e4066", "metadata": {}, - "outputs": [], "source": [ "!python -c \"import monai\" || pip install -q \"monai-weekly[pillow, ignite, tqdm]\"\n", "!python -c \"import matplotlib\" || pip install -q matplotlib\n", @@ -419,7 +418,7 @@ "source": [ "max_epochs = 5\n", "\n", - "pretrained_model = monai.bundle.load(name=\"endoscopic_inbody_classification\", bundle_dir=\"./\", return_state_dict=False)\n", + "pretrained_model = monai.bundle.load(name=\"endoscopic_inbody_classification\", bundle_dir=\"./\")\n", "\n", "pretrained_model.train()\n", "losses = []\n", @@ -483,7 +482,7 @@ ], "metadata": { "kernelspec": { - "display_name": "Python 3 (ipykernel)", + "display_name": "Python 3", "language": "python", "name": "python3" }, @@ -497,7 +496,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.8.10" + "version": "3.10.12" } }, "nbformat": 4,