patrickvonplaten commited on
Commit
b2d5b71
·
1 Parent(s): 9370a9e

Update config.json

Browse files
Files changed (1) hide show
  1. 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
  }