gaurav pathak
Results
2
comments of
gaurav pathak
trafficstars
When you load the data with the "ToTensor()" transform on as is done by the author, it loads the data as BCHW, so the code is correct.
Any update on this, I am still facing this problem on AWS instance