BackTo2014 commited on
Commit
63c3882
1 Parent(s): 893185d

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +8 -8
config.json CHANGED
@@ -2,16 +2,16 @@
2
  "in_channels": 3,
3
  "out_channels": 3,
4
  "down_block_types": [
5
- ["ResBlock", "ResBlock", "DownSample"],
6
- ["ResBlock", "AttnResBlock", "ResBlock", "DownSample"],
7
- ["ResBlock", "ResBlock", "DownSample"],
8
- ["ResBlock", "ResBlock"]
9
  ],
10
  "up_block_types": [
11
- ["ResBlock", "ResBlock"],
12
- ["ResBlock", "ResBlock", "DownSample"],
13
- ["ResBlock", "AttnResBlock", "ResBlock", "DownSample"],
14
- ["ResBlock", "ResBlock", "DownSample"]
15
  ],
16
  "block_out_channels": [
17
  128, 256, 256, 256
 
2
  "in_channels": 3,
3
  "out_channels": 3,
4
  "down_block_types": [
5
+ "DownSample",
6
+ "DownSample",
7
+ "DownSample",
8
+ "ResBlock"
9
  ],
10
  "up_block_types": [
11
+ "ResBlock",
12
+ "DownSample",
13
+ "DownSample",
14
+ "DownSample"
15
  ],
16
  "block_out_channels": [
17
  128, 256, 256, 256