LittlePey / SFD

Sparse Fuse Dense: Towards High Quality 3D Detection with Depth Completion (CVPR 2022, Oral)
Apache License 2.0
262 stars 35 forks source link

ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 12, 1]) #46

Open ywh939 opened 1 year ago

ywh939 commented 1 year ago

您好!

我是一名AI学习新手,在一张2070Super复现您代码的过程中,出现了这个问题。

第一次是在第一个epoch快结束时出现,网上查了些,试了batch_size调大,但调大有另外的问题,试了设置drop_last=True,也有其他的问题。

在刚刚按最初的配置运行,完成一次epoch,以为这个问题没有了,结果第二次epoch开始没多久,又出现了。

微信截图_20230306202117 微信截图_20230306201017 微信截图_20230306201049
jellywangjie commented 1 year ago

请问解决了吗,我也有同样的问题

ywh939 commented 1 year ago

请问解决了吗,我也有同样的问题

没有解决。但我在出现这个问题后,继续输入同样的命令进行train,出现一两次后,发现当train 3个epoch后就再也没有跳出了,而且正常train结束,结果也和论文的差不多。

jellywangjie commented 1 year ago

非常感谢

HuangLLL123 commented 7 months ago

请问解决了吗,我也有同样的问题

没有解决。但我在出现这个问题后,继续输入同样的命令进行train,出现一两次后,发现当train 3个epoch后就再也没有跳出了,而且正常train结束,结果也和论文的差不多。

请问你这里有对原来的代码或config进行修改吗,还是按最初的配置不断继续训练?

HuangLLL123 commented 7 months ago

朋友,请问你后面找到有用的解决方法了吗?我一直卡在第2个epoch,无法进入第3个以及后面的epoch

ywh939 commented 7 months ago

印象中是没有修改的。因为该环境不能在30系显卡运行,后来就没再继续研究了。

你可以多试试,它报错后继续运行。或者试试多卡训练。

HuangLLL123 commented 7 months ago

印象中是没有修改的。因为该环境不能在30系显卡运行,后来就没再继续研究了。

你可以多试试,它报错后继续运行。或者试试多卡训练。

它报错后我继续运行了好多次了,一直无法突破第2个epoch进入第3个,一直在不同的样本报错,不过还是谢谢你