ethliup / DeepUnrollNet

Deep Shutter Unrolling Network
MIT License
44 stars 6 forks source link

About hyper-parameters #1

Open zzh-tech opened 3 years ago

zzh-tech commented 3 years ago

I found that the default hyperparameters in your repository are not consistent with what are described in the paper. Specifically, the crop size and the number of epoch in your repository are set to 256x448 and 1000 respectively, while in the paper they are 320x256 and 200. What are the correct hyperparameters for reproducing the results?

ethliup commented 3 years ago

Sorry for my late reply. If I remember correctly, you can use 320x256 for 200 epochs. I can reconfirm with you after CVPR deadline if necessary, by retraining it. In case there is anything broken with the training scripts.