Axes don't match array error in pytorch
I was trying to run the vnet implementation in pytorch
(https://github.com/mattmacy/vnet.pytorch) and after normalising the scans with 

x_max = 512 
y_max = 512 
z_max = 500 
voxspacing = 0.7 


when I call the tran function on the line where the for loop is enumerating through the data loader I get a 


  axes don't match array error. 


I don't understand how to fix this. Any help would be really helpful. I am a newbie in machine learning

comment the lines 417 and 418. the issue will get fixed

the issue is because of these 2 lines

if self.transform is not None:
     img = self.transform(img)

