Commit
·
2503402
1
Parent(s):
8236408
Update config.json
Browse files- config.json +2 -2
config.json
CHANGED
|
@@ -53,22 +53,22 @@
|
|
| 53 |
"resnet_num_groups": null,
|
| 54 |
"down_blocks": [
|
| 55 |
"UNetResSkipDownBlock2D",
|
| 56 |
-
"UNetResAttnSkipDownBlock2D",
|
| 57 |
"UNetResSkipDownBlock2D",
|
| 58 |
"UNetResSkipDownBlock2D",
|
| 59 |
"UNetResSkipDownBlock2D",
|
| 60 |
"UNetResSkipDownBlock2D",
|
| 61 |
"UNetResSkipDownBlock2D",
|
|
|
|
| 62 |
"UNetResSkipDownBlock2D"
|
| 63 |
],
|
| 64 |
"up_blocks": [
|
| 65 |
"UNetResSkipUpBlock2D",
|
|
|
|
| 66 |
"UNetResSkipUpBlock2D",
|
| 67 |
"UNetResSkipUpBlock2D",
|
| 68 |
"UNetResSkipUpBlock2D",
|
| 69 |
"UNetResSkipUpBlock2D",
|
| 70 |
"UNetResSkipUpBlock2D",
|
| 71 |
-
"UNetResAttnSkipUpBlock2D",
|
| 72 |
"UNetResSkipUpBlock2D"
|
| 73 |
]
|
| 74 |
}
|
|
|
|
| 53 |
"resnet_num_groups": null,
|
| 54 |
"down_blocks": [
|
| 55 |
"UNetResSkipDownBlock2D",
|
|
|
|
| 56 |
"UNetResSkipDownBlock2D",
|
| 57 |
"UNetResSkipDownBlock2D",
|
| 58 |
"UNetResSkipDownBlock2D",
|
| 59 |
"UNetResSkipDownBlock2D",
|
| 60 |
"UNetResSkipDownBlock2D",
|
| 61 |
+
"UNetResAttnSkipDownBlock2D",
|
| 62 |
"UNetResSkipDownBlock2D"
|
| 63 |
],
|
| 64 |
"up_blocks": [
|
| 65 |
"UNetResSkipUpBlock2D",
|
| 66 |
+
"UNetResAttnSkipUpBlock2D",
|
| 67 |
"UNetResSkipUpBlock2D",
|
| 68 |
"UNetResSkipUpBlock2D",
|
| 69 |
"UNetResSkipUpBlock2D",
|
| 70 |
"UNetResSkipUpBlock2D",
|
| 71 |
"UNetResSkipUpBlock2D",
|
|
|
|
| 72 |
"UNetResSkipUpBlock2D"
|
| 73 |
]
|
| 74 |
}
|