SFD icon indicating copy to clipboard operation
SFD copied to clipboard

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

Open ywh939 opened this issue 1 year ago • 7 comments

您好!

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

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

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

微信截图_20230306202117 微信截图_20230306201017 微信截图_20230306201049

ywh939 avatar Mar 06 '23 12:03 ywh939

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

jellywangjie avatar Mar 28 '23 15:03 jellywangjie

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

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

ywh939 avatar Mar 29 '23 02:03 ywh939

非常感谢

jellywangjie avatar Mar 30 '23 10:03 jellywangjie

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

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

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

HuangLLL123 avatar Nov 22 '23 13:11 HuangLLL123

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

HuangLLL123 avatar Nov 22 '23 13:11 HuangLLL123

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

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

ywh939 avatar Nov 22 '23 13:11 ywh939

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

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

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

HuangLLL123 avatar Nov 22 '23 14:11 HuangLLL123