diff --git a/ppgan/models/pix2pix_model.py b/ppgan/models/pix2pix_model.py
index c68926fdd20ade2f1d4cc02297371ac42fd3eacd..e98039120aa6fc2467dbedc891e0c3c6b206012f 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']