From 42a445360e4c28969729cb76f60c30d04b53626b Mon Sep 17 00:00:00 2001
From: LielinJiang <jianglielin@baidu.com>
Date: Thu, 2 Jul 2020 03:06:47 +0000
Subject: [PATCH] support dict dataset

---
 ppgan/datasets/builder.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/ppgan/datasets/builder.py b/ppgan/datasets/builder.py
index e94ceba..7dc8be5 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)
     
-- 
GitLab