RainNet icon indicating copy to clipboard operation
RainNet copied to clipboard

关于总的epoch数

Open YCHang686 opened this issue 2 years ago • 16 comments

论文中总epoch数为100,但默认option中总epoch数为100 + 100 = 200。请问是否应该设置为50 + 50?

YCHang686 avatar Jul 22 '21 09:07 YCHang686

我本来是100+100的,但是我发现吧,恒定学习率训100epoch也可以满足我的目的了,而200epochs需要多一倍的时间。所以你可以试试这种50+50这种调参方式,后面50epoch学习率会衰减到0,可能不如恒定学习率稳定,不过还是建议你试一试吧

junleen avatar Jul 22 '21 09:07 junleen

感谢您的回复,还想再请教您一下: 1.训练的时候preprocess这个选项对结果影响大么?因为我看您的更新,强调了使用flip和crop; 2.目前程序中preprocess是不是不含flip,需要自己改写程序?

YCHang686 avatar Jul 25 '21 07:07 YCHang686

我的实验中都是没有数据增强的,都没有flip和crop,所有的实验都是保持了同一设置。新版的是512的分辨率,所以用了,也是试试看。另外,如果你用数据增强的话,建议所有的实验包括消融实验都需要保持同样的数据增强设置,不然对照实验没法做

junleen avatar Jul 25 '21 07:07 junleen

好嘞,谢谢回复!

YCHang686 avatar Jul 25 '21 11:07 YCHang686

您好,我下载了您提供的checkpoint,但使用evaluate.py测试时,最终结果为34.7762dB,论文中为36.12dB,请问这是什么原因呢?

YCHang686 avatar Jul 27 '21 06:07 YCHang686

没有吧,我又运行了一次,很正常哎

image

junleen avatar Jul 27 '21 06:07 junleen

感谢您的回复。 首先,我感觉最后的显示PSNR和MSE应该是反了,应为PSNR:36.0516; 然后,我这边测试依然是34.7762,请问您可以检查一下上传的evaluate.py和net_G.pth文件吗,我怀疑是不是上传错误的问题。 1

YCHang686 avatar Jul 27 '21 09:07 YCHang686

额, 是写反了

写成这个了

print('MSE: %.4f  PSNR: %.4f' % (calculateMean(all_psnr), calculateMean(all_mse)))

junleen avatar Jul 27 '21 09:07 junleen

不好意思,文件是不太一样,可以你用那个net_G_last.pth 试试

junleen avatar Jul 27 '21 09:07 junleen

嗯嗯我刚试了一下,PSNR是34.7814,应该也不是这个文件

YCHang686 avatar Jul 27 '21 10:07 YCHang686

我下载了google drive上面的模型,测试结果还是36.05,我担心是不是pytorch版本不一样导致的?我在1.7.0和1.8.1上测试结果都是一样的,你是否用的是更早期的版本呢?

junleen avatar Jul 27 '21 10:07 junleen

我用的是pytorch1.6,那我试一试1.7

YCHang686 avatar Jul 27 '21 11:07 YCHang686

我用的是pytorch1.6,那我试一试1.7

搞定了吗?

junleen avatar Jul 27 '21 11:07 junleen

嗯嗯已解决,感谢您的回复!

YCHang686 avatar Jul 28 '21 03:07 YCHang686

好嘞,谢谢回复!

@junleen 您好,我也遇到了相同的问题,我更换了pytorch 版本还是不可以。请问您是怎么解决的

tanjuju avatar Aug 11 '22 14:08 tanjuju

没有吧,我又运行了一次,很正常哎

image

@YCHang686 您好,我也下载了您提供的checkpoint,但使用evaluate.py测试时,也换了不同版本的pytorch,但最终结果为35.887与论文中为36.12dB不同,请问这是什么原因呢? image

tanjuju avatar Aug 11 '22 14:08 tanjuju