patrickvonplaten commited on
Commit
2503402
·
1 Parent(s): 8236408

Update config.json

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