From 05e26e8d05b202940b3a3d194a1f91cb4d62178a Mon Sep 17 00:00:00 2001
From: xiexionghang <xiexionghang@baidu.com>
Date: Thu, 1 Aug 2019 18:36:46 +0800
Subject: [PATCH] add dataset

---
 paddle/fluid/train/custom_trainer/feed/dataset/data_reader.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/paddle/fluid/train/custom_trainer/feed/dataset/data_reader.h b/paddle/fluid/train/custom_trainer/feed/dataset/data_reader.h
index d7841d35..5856cebc 100644
--- a/paddle/fluid/train/custom_trainer/feed/dataset/data_reader.h
+++ b/paddle/fluid/train/custom_trainer/feed/dataset/data_reader.h
@@ -8,6 +8,7 @@
 #include <memory>
 #include <yaml-cpp/yaml.h>
 #include "paddle/fluid/framework/channel.h"
+#include "paddle/fluid/train/custom_trainer/feed/common/registerer.h"
 
 namespace paddle {
 namespace custom_trainer {
-- 
GitLab