Commit
·
b2d5b71
1
Parent(s):
9370a9e
Update config.json
Browse files- config.json +2 -1
config.json
CHANGED
|
@@ -71,5 +71,6 @@
|
|
| 71 |
"UNetResSkipUpBlock2D",
|
| 72 |
"UNetResSkipUpBlock2D"
|
| 73 |
],
|
| 74 |
-
"center_input_sample": true
|
|
|
|
| 75 |
}
|
|
|
|
| 71 |
"UNetResSkipUpBlock2D",
|
| 72 |
"UNetResSkipUpBlock2D"
|
| 73 |
],
|
| 74 |
+
"center_input_sample": true,
|
| 75 |
+
"num_head_channels": 1,
|
| 76 |
}
|