add dataset

05e26e8d · xiexionghang · 147fe8cb · 05e26e8d
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/train/custom_trainer/feed/dataset/data_reader.h paddle/fluid/train/custom_trainer/feed/dataset/data_reader.h +1 -0

未找到文件。
--- 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 {