train.py文件的第98行代码的疑惑

[hpc203]

看train.py文件的第98行代码： image = torch.FloatTensor(opt.batchSize, 3, opt.imgH, opt.imgH) 按照常理来说，应该是image = torch.FloatTensor(opt.batchSize, 3, opt.imgH, opt.imgW) 可是我改成这样之后重新训练，不收敛的。请问作者，这里的image为什么要定义成torch.FloatTensor(opt.batchSize, 3, opt.imgH, opt.imgH)而不是torch.FloatTensor(opt.batchSize, 3, opt.imgH, opt.imgW)呢？

[lmw0320]

我也很奇怪。。。这里不应该是imgH, imgW么，为什么会是两个imgH呢？？