PolynomialQian / RD3D

RD3D: RGB-D Salient Object Detection via 3D Convolutional Neural Networks
MIT License
49 stars 8 forks source link

关于模型训练的问题 #4

Open ltqqq opened 3 years ago

ltqqq commented 3 years ago

作者你好,我这几天使用您提供的代码重新训练了好几次,我使用的是单张TitanXp,batch_size设为8(10的话显卡放不下),epochs=100,其他参数基本都使用您代码中的默认参数,最终训练loss都收敛到0.03左右,但测试效果都很不理想。所以我想请教一下您训练时的具体训练参数和最终loss收敛值是多少,最好能提供一下训练好的模型文件? 万分感谢!

PolynomialQian commented 3 years ago

你好,感谢关注我们的项目,具体的训练参数在论文中均已提供,我们是用4张Titan Xp训练的,并非单张显卡。我们会在近期提供训练好的model。

在 2021-04-03 22:25:49,"llltttqq" @.***> 写道:

作者你好,我这几天使用您提供的代码重新训练了好几次,我使用的是单张TitanXp,batch_size设为8(10的话显卡放不下),epochs=100,其他参数基本都使用您代码中的默认参数,最终训练loss都收敛到0.03左右,但测试效果都很不理想。所以我想请教一下您训练时的具体训练参数和最终loss收敛值是多少,最好能提供一下训练好的模型文件? 万分感谢!

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or unsubscribe.

ltqqq commented 3 years ago

谢谢,想顺便问一下,在论文中你们提到,当在其它数据集进行测试时用的是1485NJU2K+700NLPR数据训练的模型,而在DUT数据集上测试时用的是额外加入800DUT训练数据训练的模型,这种做法是现在RGBD SOD的普遍做法吗?因为许多用到DUT进行训练的方法都没有详细提到这一点,只是单纯提到训练集由1485NJU2K+700NLPR+800DUT组成

PolynomialQian commented 3 years ago

是的

在 2021-04-04 22:04:31,"llltttqq" @.***> 写道:

谢谢,想顺便问一下,在论文中你们提到,当在其它数据集进行测试时用的是1485NJU2K+700NLPR数据训练的模型,而在DUT数据集上测试时用的是额外加入800DUT训练数据训练的模型,这种做法是现在RGBD SOD的普遍做法吗?因为许多用到DUT进行训练的方法都没有详细提到这一点,只是单纯提到训练集由1485NJU2K+700NLPR+800DUT组成

— You are receiving this because you commented. Reply to this email directly, view it on GitHub, or unsubscribe.

ltqqq commented 3 years ago

谢谢

clelouch commented 3 years ago

如果采用单卡训练,似乎应该调整学习率,从40的batch size到8大概学习率除以2可能会有效果