Update unet/config.json
Browse files- unet/config.json +1 -2
unet/config.json
CHANGED
|
@@ -20,6 +20,5 @@
|
|
| 20 |
"num_head_channels": 64,
|
| 21 |
"num_res_blocks": 2,
|
| 22 |
"out_channels": 4,
|
| 23 |
-
"transformer_depth": 1
|
| 24 |
-
"use_checkpoint": false
|
| 25 |
}
|
|
|
|
| 20 |
"num_head_channels": 64,
|
| 21 |
"num_res_blocks": 2,
|
| 22 |
"out_channels": 4,
|
| 23 |
+
"transformer_depth": 1
|
|
|
|
| 24 |
}
|