Open Hknight514880265 opened 4 years ago
Hello, thanks for your great work. Hope you can answer my questions: when you create "image" Tensor,it's shape is [3, imgH, imgH] why not [3, imgH, imgW]?
Thank you!
details in train.py line 122.
image = torch.FloatTensor(params.batchSize, 3, params.imgH, params.imgH)
[3, imgH, imgW] or [3, imgH, imgH], do you kown which is right?
Hello, thanks for your great work. Hope you can answer my questions: when you create "image" Tensor,it's shape is [3, imgH, imgH] why not [3, imgH, imgW]?
Thank you!