Skip to content

Commit 2d3b205

Browse files
add ignite (#753)
Signed-off-by: Yiheng Wang <vennw@nvidia.com>
1 parent c3a7e95 commit 2d3b205

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

modules/bundle/spleen_segmentation/configs/metadata.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,8 @@
99
"pytorch_version": "1.10.0",
1010
"numpy_version": "1.21.2",
1111
"optional_packages_version": {
12-
"nibabel": "3.2.1"
12+
"nibabel": "3.2.1",
13+
"pytorch-ignite": "0.4.8"
1314
},
1415
"task": "Decathlon spleen segmentation",
1516
"description": "A pre-trained model for volumetric (3D) segmentation of the spleen from CT image",

0 commit comments

Comments
 (0)