gangweiX/IGEV

自己重新训练数据集时出现问题

Yu-Cheung opened this issue · 3 comments

Yu-Cheung commented

我用 $Scene \ Flow$ 数据集从零开始训练， $Loss$ 一直在上升，同时 $EPE$ 也在上升，并且 $Loss$ 和 $EPE$ 在最后都会直接收敛。是数据集出现问题了吗？然后这是我训练时候的 $Loss$ 和 $EPE$。对于整个网络，我只修改了数据集读取的路径，是不是我的数据集本身出现问题了？
image
image

gangweiX commented

不明白，我从来没有出现过这种情况。你一直都是这种情况嘛？

Windsrain commented

我重新从github上下载了最新代码进行训练，目前跑了5k steps没有任何问题。
JT{54%EYJ0_X0XG %O50L

你确定没有修改任何源码么？

Yu-Cheung commented

我找到原因了，torch的版本不对，不好意思，浪费了大家的时间