Hello ptrblck,Thank you for your time
I tried doing torch.max(output,1, keepdim = True),after converting it to numpy when i do np.unique(), i only get [0],surprisingly when i set the batch size to 2 and print out output.size() i get [1,1,128,128],I was expecting [2,1,128,128],Anyways do you have any suggestions on this?
Thanks