Hello,
I am trying to get my net to output a prediction for each input image, but I am only getting it to predict two value, rather then 4 for each input image.
image1135×752 78.1 KB
I was going through a pretrained CNN (mnist for digits) and it predicts 4 tensors desptie its final linear layer outputting 10 (not 4)
mnist dataset - not the one I want to fix
image1209×720 76.3 KB