Closed 5ummer23 closed 1 week ago
作者您好打扰您了,我跑通了您的代码,尝试跑一下mind101数据集,请问您可以分享一下大致跑了多少轮后差不多达到最优吗?我看您在设置参数时default给了1000,由于算力有限我先用了NKI-RS-22这22张尝试跑一下,恳请您指教一下训练过程中是否有不妥的地方,十分感谢!
看起来没什么问题,不过实际上22张实在是太少了,最好按照我们在论文中提供的配置重新分一下数据集,实际上无论数据集多大每一个epoch的step都设置成了100,所以没有任何影响。训练效果你可以看第一项的ncc loss 的收敛情况,大概200-300轮就有一个相对稳定的结果了。
最优的结果我是在最后900-1000epoch里选择的验证集上效果最好的模型。
作者您好打扰您了,我跑通了您的代码,尝试跑一下mind101数据集,请问您可以分享一下大致跑了多少轮后差不多达到最优吗?我看您在设置参数时default给了1000,由于算力有限我先用了NKI-RS-22这22张尝试跑一下,恳请您指教一下训练过程中是否有不妥的地方,十分感谢!