torch.onnx.errors.OnnxExporterError: Module onnx is not installed!
安装python包 onnx
pip install onnx
ValueError: operands could not be broadcast together with shapes。。。
我是在训练分割模型,出现这个问题是矩阵相乘时不匹配宽高,发生冲突。
我检查了我的标签图片,0为背景,1为分割类别,却出现了2的值,所以2和3冲突,改一下制作标签的代码就好。