update details

Author: IceClear

README.md

@@ -75,8 +75,8 @@ Here the models we provide are trained on QP37 in RGB space.
 
 :star: *MWGAN Model:*
 
-- [MWGAN PSNR Model](https://drive.google.com/u/0/uc?id=1lvki-CphYSVvnw576BkUzyX_dqJTN7g8&export=download)
-- [MWGAN GAN Model]() (In progress)
+- [MWGAN PSNR Model](https://drive.google.com/u/0/uc?id=1lvki-CphYSVvnw576BkUzyX_dqJTN7g8&export=download): Pretrained PSNR model to initialize generator for GAN training.
+- [MWGAN GAN Model](): The model for MW-GAN. (In progress)
 
 ## Acknowledgement
 This repo is built mainly based on [BasicSR](https://github.com/xinntao/BasicSR). Also borrowing codes from [pacnet](https://github.com/NVlabs/pacnet) and [MWCNN_PyTorch](https://github.com/lpj0/MWCNN_PyTorch). We thank a lot for their contributions to the community.

options/test/MWGAN/test_MWGAN_ECCV_PSNR.yml

@@ -28,7 +28,7 @@ network_g:
 
 # path
 path:
-  pretrain_network_g: ./experiments/pretrain_model/MWGAN_ECCV_PSNR_model.pth
+  pretrain_network_g: ./experiments/pretrain_model/MWGAN_PSNR_ECCV_model.pth
   strict_load_g: True
 
 # validation settings

options/test/MWGAN/test_MWGAN_ECCV_Percep.yml

@@ -28,7 +28,7 @@ network_g:
 
 # path
 path:
-  pretrain_network_g: ./experiments/pretrain_model/MWGAN_Perep_model.pth
+  pretrain_network_g: ./experiments/pretrain_model/MWGAN_Perep_ECCV_model.pth
   strict_load_g: true
 
 # validation settings