Fix deepedit tutorial (#696)

diazandr3s · web-flow · commit 3c1629052645 · 2022-05-11T16:52:50.000+01:00
Signed-off-by: Andres Diaz-Pinto &lt;diazandr3s@gmail.com&gt;
diff --git a/deepedit/ignite/infoANDinference.ipynb b/deepedit/ignite/infoANDinference.ipynb
@@ -27,7 +27,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!python -c \"import monai\" || pip install -q \"monai-weekly[nibabel tqdm]\"\n",
+    "!python -c \"import monai\" || pip install -q \"monai-weekly[nibabel, tqdm]\"\n",
     "!python -c \"import matplotlib\" || pip install -q matplotlib==3.5.2\n",
     "!pip install -q pytorch-ignite==0.4.8\n",
     "\n",
diff --git a/deepedit/ignite/train.py b/deepedit/ignite/train.py
@@ -351,7 +351,7 @@ def run(args):
             "{}:: Loading PT Model from: {}".format(args.local_rank, args.input)
         )
         device = torch.device("cuda" if args.use_gpu else "cpu")
-        network = get_network(args.network, args.channels, args.dimensions).to(device)
+        network = get_network(args.network, args.labels, args.spatial_size).to(device)
 
         map_location = {"cuda:0": "cuda:{}".format(args.local_rank)}
         network.load_state_dict(torch.load(args.input, map_location=map_location))

Original file line number	Diff line number	Diff line change
`@@ -351,7 +351,7 @@ def run(args):`
`351`	`351`	`"{}:: Loading PT Model from: {}".format(args.local_rank, args.input)`
`352`	`352`	`)`
`353`	`353`	`device = torch.device("cuda" if args.use_gpu else "cpu")`
`354`		`- network = get_network(args.network, args.channels, args.dimensions).to(device)`
	`354`	`+ network = get_network(args.network, args.labels, args.spatial_size).to(device)`
`355`	`355`
`356`	`356`	`map_location = {"cuda:0": "cuda:{}".format(args.local_rank)}`
`357`	`357`	`network.load_state_dict(torch.load(args.input, map_location=map_location))`