Update options/train/LOLBlur.yml

options/train/LOLBlur.yml

@@ -30,23 +30,24 @@ network:
   name: Network
   img_channels: 3
   width: 32
-  middle_blk_num: 3
+  middle_blk_num_enc: 2
+  middle_blk_num_dec: 2
   enc_blk_nums: [1, 2, 3]
   dec_blk_nums: [3, 1, 1]
   enc_blk_nums_map: None
   middle_blk_num_map: None
   residual_layers: None
-  dilations: [1, 4]
+  dilations: [1, 4, 9]
   spatial: None
   extra_depth_wise: True
 
 
 #### training settings: learning rate scheme, loss
 train:
-  lr_initial: !!float 2e-5
+  lr_initial: !!float 5e-4
   lr_scheme: CosineAnnealing
   betas: [0.9, 0.9]
-  epochs: 700
+  epochs: 500
   lr_gamma: 0.5
   weight_decay: !!float 1e-3
   eta_min: !!float 1e-6
@@ -71,15 +72,15 @@ train:
 
 #### save model
 save:
-  path: ./models/Network_v3_new_network_noFPA_LOLBlur.pt
+  path: ./models/Network_noFAC_LOLBlur.pt
   best: ./models/bests/
-  new: ./models/Network_v3_new_network_noFPA_LOLBlur_v2.pt 
+  new: ./models/Network_noFAC_LOLBlur.pt 
 #### wandb:
 wandb:
   init: True
   project: LOLBlur
   entity: cidautai
-  name: Network_v3_new_network_noFPA_LOLBlur
+  name: Network_v3_noFAC_LOLBlur
   save_code: True