Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,8 @@ pil_to_tensor = transforms.ToTensor()
|
|
23 |
|
24 |
model = Network(img_channel=opt['network']['img_channels'],
|
25 |
width=opt['network']['width'],
|
26 |
-
|
|
|
27 |
enc_blk_nums=opt['network']['enc_blk_nums'],
|
28 |
dec_blk_nums=opt['network']['dec_blk_nums'],
|
29 |
dilations=opt['network']['dilations'],
|
|
|
23 |
|
24 |
model = Network(img_channel=opt['network']['img_channels'],
|
25 |
width=opt['network']['width'],
|
26 |
+
middle_blk_num_enc=opt['network']['middle_blk_num_enc'],
|
27 |
+
middle_blk_num_dec=opt['network']['middle_blk_num_dec'],
|
28 |
enc_blk_nums=opt['network']['enc_blk_nums'],
|
29 |
dec_blk_nums=opt['network']['dec_blk_nums'],
|
30 |
dilations=opt['network']['dilations'],
|