update details

IceClear · Nov 17, 2021 · d71fb71 · d71fb71
1 parent 7eea517
commit d71fb71
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/README.md b/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.

diff --git a/options/test/MWGAN/test_MWGAN_ECCV_PSNR.yml b/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

diff --git a/options/test/MWGAN/test_MWGAN_ECCV_Percep.yml b/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