基于paddle2.2 Api使用U-Net网络实现图像的语义分割
收藏
开始训练时报如下错误，请问如何改正？
ValueError: Target(Tensor(shape=[1], dtype=int64, place=CUDAPlace(0), stop_gradient=True,
[3])) is out of class_dimension's upper bound(1)
奔向未来的样子
已解决
3#
回复于2022-01
明显的维度问题 如果你用的是别人的代码，建议不要改变size等设置，把自己的数据扔进去试试能不能跑得起来，有时网络太大数据太小，给卷没了，或者三通道四通道等问题，得具体分析。 如果你的代码是自己写的，追一下是哪个层报错，看看维度是不是对齐了。
0
收藏
请登录后评论
你的target输出的是一个值？应该是shape和input相同的特征吧？检查下维度，可以用paddle.summary检查下维度，具体怎么用可以在官网查
明显的维度问题
如果你用的是别人的代码，建议不要改变size等设置，把自己的数据扔进去试试能不能跑得起来，有时网络太大数据太小，给卷没了，或者三通道四通道等问题，得具体分析。
如果你的代码是自己写的，追一下是哪个层报错，看看维度是不是对齐了。
fork的别人的代码，大概听懂了，我再试试，谢谢