Torbjorn1997 / IIRP-Net

8 stars 0 forks source link

关于训练的问题 #4

Closed 5ummer23 closed 1 week ago

5ummer23 commented 1 week ago

作者您好打扰您了,我跑通了您的代码,尝试跑一下mind101数据集,请问您可以分享一下大致跑了多少轮后差不多达到最优吗?我看您在设置参数时default给了1000,由于算力有限我先用了NKI-RS-22这22张尝试跑一下,恳请您指教一下训练过程中是否有不妥的地方,十分感谢!

截屏2024-09-12 11 44 40
Torbjorn1997 commented 1 week ago

看起来没什么问题,不过实际上22张实在是太少了,最好按照我们在论文中提供的配置重新分一下数据集,实际上无论数据集多大每一个epoch的step都设置成了100,所以没有任何影响。训练效果你可以看第一项的ncc loss 的收敛情况,大概200-300轮就有一个相对稳定的结果了。

Torbjorn1997 commented 1 week ago

最优的结果我是在最后900-1000epoch里选择的验证集上效果最好的模型。