From 1c5c2f50da5ae101077c27fdac2a12fb1619ec86 Mon Sep 17 00:00:00 2001 From: SsnL Date: Sat, 13 Jan 2018 23:04:43 -0500 Subject: fix resize_ issue #170 --- models/test_model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'models/test_model.py') diff --git a/models/test_model.py b/models/test_model.py index 2ae2812..f593c46 100644 --- a/models/test_model.py +++ b/models/test_model.py @@ -12,8 +12,6 @@ class TestModel(BaseModel): def initialize(self, opt): assert(not opt.isTrain) BaseModel.initialize(self, opt) - self.input_A = self.Tensor(opt.batchSize, opt.input_nc, opt.fineSize, opt.fineSize) - self.netG = networks.define_G(opt.input_nc, opt.output_nc, opt.ngf, opt.which_model_netG, opt.norm, not opt.no_dropout, @@ -29,7 +27,9 @@ class TestModel(BaseModel): def set_input(self, input): # we need to use single_dataset mode input_A = input['A'] - self.input_A.resize_(input_A.size()).copy_(input_A) + if len(self.gpu_ids) > 0: + input_A = input_A.cuda(self.gpu_ids[0], async=True) + self.input_A = input_A self.image_paths = input['A_paths'] def test(self): -- cgit v1.2.3-70-g09d2