fixed the dimension in convolution according to MONAI 1.0 update
Browse files- configs/metadata.json +2 -1
- scripts/networks/unest.py +1 -1
configs/metadata.json
CHANGED
@@ -1,7 +1,8 @@
|
|
1 |
{
|
2 |
"schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20220324.json",
|
3 |
-
"version": "0.1.
|
4 |
"changelog": {
|
|
|
5 |
"0.1.1": "fixed the model state dict name",
|
6 |
"0.1.0": "complete the model package"
|
7 |
},
|
|
|
1 |
{
|
2 |
"schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20220324.json",
|
3 |
+
"version": "0.1.2",
|
4 |
"changelog": {
|
5 |
+
"0.1.2": "fixed the dimension in convolution according to MONAI 1.0 update",
|
6 |
"0.1.1": "fixed the model state dict name",
|
7 |
"0.1.0": "complete the model package"
|
8 |
},
|
scripts/networks/unest.py
CHANGED
@@ -194,7 +194,7 @@ class UNesT(nn.Module):
|
|
194 |
)
|
195 |
|
196 |
self.encoder10 = Convolution(
|
197 |
-
|
198 |
in_channels=32 * feature_size,
|
199 |
out_channels=64 * feature_size,
|
200 |
strides=2,
|
|
|
194 |
)
|
195 |
|
196 |
self.encoder10 = Convolution(
|
197 |
+
spatial_dims=3,
|
198 |
in_channels=32 * feature_size,
|
199 |
out_channels=64 * feature_size,
|
200 |
strides=2,
|