diff --git a/ppgan/datasets/builder.py b/ppgan/datasets/builder.py
index e94cebafc5884f0921b3e2f8dffa6146a5350172..7dc8be530d0be6e1ba48fa7007208c4e8e9b842a 100644
--- a/ppgan/datasets/builder.py
+++ b/ppgan/datasets/builder.py
@@ -97,7 +97,7 @@ class DictDataLoader():
         if isinstance(indexs, paddle.Variable):
             indexs = indexs.numpy()
         current_items = []
-        items = getattr(self.dataset, key)
+        items = self.dataset.non_tensor_dict[key]
 
         for index in indexs:
             current_items.append(items[index])
@@ -105,7 +105,6 @@ class DictDataLoader():
         return current_items
 
 
-
 def build_dataloader(cfg, is_train=True):
     dataset = DATASETS.get(cfg.name)(cfg)