咨询

[hhjiann]

作者你好，我在复现时遇到了问题。具体问题是有关你的网络结构在0.3bpp嵌入率及以下嵌入率时，存在不收敛的问题。我想咨询一下有关你的网络参数初始化的方法是什么？这是非常重要，它直接影响网络的收敛问题。

[henryccl]

在FPNet中网络参数初始化使用的是torch自带的初始化，如果你的网络没有收敛，你可以从如下几个方面考虑：

• 数据集是否正确（信息嵌入时是否有问题？）
• 对不同嵌入率的数据集调整学习率！！！！
• 使用计算精度更高的显卡，如v100等（先前有关项目明确讨论过这个问题，在这个任务上浮点数计算精度会影响收敛，很抱歉我找不到具体链接了，推荐看看SRNet和SiastegNet的相关实现，或是文章细节）
• 一般而言模型不应该太深
• 除SRNet之外，推荐使用先验信息引导网络收敛（如kv核等）
• 优化流程是否合理？（推荐参考SRNet的优化流程https://github.com/brijeshiitg/Pytorch-implementation-of-SRNet）