But as far as I can see you have a fc layer which goes from 256 to 1 feature. I find it’s not possible you to get 195 elements on a tensor. If you does not modify the network it should be a tensor of size batch x 1 or similar to that.
Could you print output shape? Or, are you running batch size 195 or getting 195 items in the batch dimension when you apply that x = x.view(-1, 32 * 4 * 4)?