-
Notifications
You must be signed in to change notification settings - Fork 46
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
模型参数大小与论文不符 #18
Comments
你好请问你训练出来的结果有论文上的那么高吗 |
我不太清楚作者使用的可变形卷积是哪个版本,我试过几个但是没有编译成功,不使用DCN的情况下,在ITS数据集中我训练出来的PSNR结果大概是31左右 |
https://github.com/jinfagang/DCNv2_latest 这个版本的DCN应该可以编译成功 |
本人小白,请问是在什么系统运行的,我用window pycharm运行好多错误 |
我用了MMCV的DCN替代,训练出来PSNR只有20多。。。。。 |
请问你用的哪个数据集训练的?我使用的ots(29万张),还在跑,使用的是MMCV。跑过代码都花了挺多时间(对比损失函数、option参数、DCN导包这些地方多多少少需要改,作者直接提供的代码跑不通。) |
收到回复。我还在跑,还不知道结果。我其实不太理解什么是权重共享,我打印出来模型的参数是Total_params: 2593228,如果您方便的话,是否可以私聊你问这个问题。 |
对比FFA和AECR的model.py我看懂了。权重共享的意思就是,只实例化一次DehazeBlock,但是,重复使用了6次(减少参数,才能权重共享。)。那您做的权重不共享的实验指标到达了多少哇? |
明白了,到时候我看看我的训练结果把。FFA我用ots(29万多张)跑psnr才16点几,难道是我的代码问题吗TAT. |
我确实是裁剪到了240,240。但是我数据增强方式和损失函数和数据集不一样。哎,因为我用FFA做对比实验,用我自己写的轮子和损失函数进行训练,只是用了FFA这个model,结果跑出来才16。我等服务器空闲了,全用FFA-master的代码跑一次我的数据集瞅瞅效果。(非常感谢您的回复,祝您情人节快乐 玫瑰.jpg) |
您论文里面提到AECR模型的参数大小为2.61M,但是我在使用您给出的网络训练出来的模型参数为10.5M,远远大于论文中的描述。。请问能否提供您的训练文件和预训练模型?
The text was updated successfully, but these errors were encountered: