-
Notifications
You must be signed in to change notification settings - Fork 316
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
请教一下训练细节 #113
Comments
you can check the code in this |
Thank you very much~ |
您好,请问您从头开始训练时,有遇到模型不收敛的问题吗?我的loss一直在1e29附近振荡 |
@kongdebug 你大概训练了多少轮? |
您好!请问一下,对于每次的迭代,是从DIV2K中随机抽取16张图像,然后随机裁剪出48x48的patch,最后执行8个方向的变换,获得训练数据吗?DIV一共有800张图像,那么一个epoch应该是800/16=50次迭代,您论文中是每200000次迭代学习率下降一半,也就是说是每4000个epoch才下降一次学习率吗?但是您的代码中默认是每200次迭代下降一次学习率,并且整个训练也只有1000次迭代。而按照1000个epoch训练下来,set5的PSNR只能达到32.2。可能我在训练细节上的理解与您有出入,还请不吝赐教。
由于pytorch版本升级,我重写了您的数据集代码,因此细节未知。
The text was updated successfully, but these errors were encountered: