diff --git a/PaddleRec/din/infer.py b/PaddleRec/din/infer.py
index 7108d0441a3c724ddda1f63ae9f896cce6ce1282..f0bbde2727a9ccc4a5f02e21982b4448f02d90b6 100644
--- a/PaddleRec/din/infer.py
+++ b/PaddleRec/din/infer.py
@@ -80,7 +80,7 @@ def infer():
 
     loader = fluid.io.DataLoader.from_generator(
         feed_list=[inference_program.block(0).var(e) for e in feed_target_names], capacity=10000, iterable=True)
-    loader.set_sample_list_generator(data_reader, places=fluid.cuda_places())
+    loader.set_sample_list_generator(data_reader, places=place)
 
     loss_sum = 0.0
     score = []
diff --git a/PaddleRec/din/train.py b/PaddleRec/din/train.py
index 1be7c6ca2e407bd8387cdf0876e2423e902b4cee..4b0b2854e7dfb3ad2189e43791d5324bf2e59300 100644
--- a/PaddleRec/din/train.py
+++ b/PaddleRec/din/train.py
@@ -96,7 +96,7 @@ def train():
 
     loader = fluid.io.DataLoader.from_generator(
         feed_list=feed_list, capacity=10000, iterable=True)
-    loader.set_sample_list_generator(data_reader, places=fluid.cuda_places())
+    loader.set_sample_list_generator(data_reader, places=place)
     if use_parallel:
         train_exe = fluid.ParallelExecutor(
             use_cuda=use_cuda, loss_name=avg_cost.name)