gangweiX / IGEV

[CVPR 2023] Iterative Geometry Encoding Volume for Stereo Matching and Multi-View Stereo

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

Yu-Cheung opened this issue · comments

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

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

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

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

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