danifei commited on
Commit
36db309
·
verified ·
1 Parent(s): 1623c10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- middle_blk_num=opt['network']['middle_blk_num'],
 
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'],