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,
|