From 8c646092283bb43a1b77a68083b228de00cedc6e Mon Sep 17 00:00:00 2001 From: "Koryukina, Valeria" Date: Thu, 13 Sep 2018 18:23:10 +0300 Subject: [PATCH] image tensor memory leak --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 3c3513a3..a21b4adf 100644 --- a/train.py +++ b/train.py @@ -95,7 +95,7 @@ def weights_init(m): crnn.load_state_dict(torch.load(opt.pretrained)) print(crnn) -image = torch.FloatTensor(opt.batchSize, 3, opt.imgH, opt.imgH) +image = torch.FloatTensor(opt.batchSize, 3, opt.imgH, opt.imgW) text = torch.IntTensor(opt.batchSize * 5) length = torch.IntTensor(opt.batchSize)