Commit ·
af53208
1
Parent(s): b87f97e
Update unet/config.json
Browse files- unet/config.json +2 -1
unet/config.json
CHANGED
|
@@ -46,5 +46,6 @@
|
|
| 46 |
"UpBlock2D",
|
| 47 |
"UpBlock2D"
|
| 48 |
],
|
| 49 |
-
"upsample_type": "conv"
|
|
|
|
| 50 |
}
|
|
|
|
| 46 |
"UpBlock2D",
|
| 47 |
"UpBlock2D"
|
| 48 |
],
|
| 49 |
+
"upsample_type": "conv",
|
| 50 |
+
"num_class_embeds": 6
|
| 51 |
}
|