Merge pull request PaddlePaddle#36 from LielinJiang/fix-to-variable

Fix to_variable
AnnaTrainingG · Oct 14, 2020 · b22c19e · b22c19e
2 parents 5500675 + 01b747c
commit b22c19e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ppgan/models/pix2pix_model.py b/ppgan/models/pix2pix_model.py
@@ -81,8 +81,8 @@ def set_input(self, input):
         """
 
         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']