IndexError: too many indices for tensor of dimension 2 #116
Description
I got this error when I was training the model.
It should have gone wrong here, prop_ The dimension of type should be 3, but the output is 2. I don't understand why this is so
“prop_ type_ var: tensor([[0, 1, 1, 1, 1, 1, 1, 2]])”
Maybe it has something to do with the size of my picture? My picture size is 340 * 256
Please help me！！！thank you！