change the network

app.py

@@ -9,10 +9,10 @@ import numpy as np
 import yaml
 from huggingface_hub import hf_hub_download
 
-from archs import Network_v3
+from archs import Network
 from options.options import parse
 
-path_opt = './options/test/LOLBlur.yml'
+path_opt = './options/predict/LOLBlur.yml'
 
 opt = parse(path_opt)
 
@@ -25,13 +25,13 @@ network = opt['network']['name']
 
 PATH_MODEL = opt['save']['path']
 
-model = Network_v3(img_channel=opt['network']['img_channels'], 
+model = Network(img_channel=opt['network']['img_channels'], 
                     width=opt['network']['width'], 
                     middle_blk_num=opt['network']['middle_blk_num'], 
                     enc_blk_nums=opt['network']['enc_blk_nums'],
                     dec_blk_nums=opt['network']['dec_blk_nums'], 
-                    residual_layers=opt['network']['residual_layers'],
-                    dilations=opt['network']['dilations'])
+                    dilations=opt['network']['dilations'],
+                    extra_depth_wise = opt['network']['extra_depth_wise'])
 
 checkpoints = torch.load(opt['save']['best'])
 # print(checkpoints)