search

meijieru / crnn.pytorch

Convolutional recurrent network in pytorch
MIT License
2.4k stars 657 forks source link

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

Open hpc203 opened 4 years ago

hpc203 commented 4 years ago

看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 commented 4 years ago

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