From 01b747cce74705f62f94c56925064c21440a96dd Mon Sep 17 00:00:00 2001 From: LielinJiang Date: Wed, 14 Oct 2020 02:36:50 +0000 Subject: [PATCH] fix variable --- ppgan/models/pix2pix_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppgan/models/pix2pix_model.py b/ppgan/models/pix2pix_model.py index c68926f..e980391 100644 --- a/ppgan/models/pix2pix_model.py +++ b/ppgan/models/pix2pix_model.py @@ -81,8 +81,8 @@ class Pix2PixModel(BaseModel): """ AtoB = self.opt.dataset.train.direction == 'AtoB' - self.real_A = paddle.to_variable(input['A' if AtoB else 'B']) - self.real_B = paddle.to_variable(input['B' if AtoB else 'A']) + self.real_A = paddle.to_tensor(input['A' if AtoB else 'B']) + self.real_B = paddle.to_tensor(input['B' if AtoB else 'A']) self.image_paths = input['A_paths' if AtoB else 'B_paths'] -- GitLab