提交 664be756 编写于 作者: B barrierye

update data_feed for add MultiSlotDataFeed

上级 bedaa615
...@@ -155,7 +155,7 @@ void MultiSlotDataFeed::Init(paddle::framework::DataFeedDesc& data_feed_desc) { ...@@ -155,7 +155,7 @@ void MultiSlotDataFeed::Init(paddle::framework::DataFeedDesc& data_feed_desc) {
finish_init_ = false; finish_init_ = false;
finish_set_filelist_ = false; finish_set_filelist_ = false;
finish_start_ = false; finish_start_ = false;
/*
if (!data_feed_desc.has_multi_slot_desc()){ if (!data_feed_desc.has_multi_slot_desc()){
LOG(ERROR) << "error: multi_slot_desc has not been set"; LOG(ERROR) << "error: multi_slot_desc has not been set";
exit(-1); exit(-1);
...@@ -178,7 +178,7 @@ void MultiSlotDataFeed::Init(paddle::framework::DataFeedDesc& data_feed_desc) { ...@@ -178,7 +178,7 @@ void MultiSlotDataFeed::Init(paddle::framework::DataFeedDesc& data_feed_desc) {
} }
} }
feed_vec_.resize(use_slots_.size()); feed_vec_.resize(use_slots_.size());
*/
finish_init_ = true; finish_init_ = true;
} }
......
...@@ -176,8 +176,8 @@ class DataFeed { ...@@ -176,8 +176,8 @@ class DataFeed {
LOG(ERROR) << "error: The function SetBufferSize is not implemented"; LOG(ERROR) << "error: The function SetBufferSize is not implemented";
exit(-1); exit(-1);
} }
virtual const std::vector<std::string>& GetAllSlots() {return all_slots_;} virtual const std::vector<std::string>& GetAllSlotAlias() {return all_slots_;}
virtual const std::vector<std::string>& GetUseSlots() {return use_slots_;} virtual const std::vector<std::string>& GetUseSlotAlias() {return use_slots_;}
std::vector<MixTensor>& GetFeedVec() {return feed_vec_;} std::vector<MixTensor>& GetFeedVec() {return feed_vec_;}
virtual void AddFeedVar(Variable* var, const std::string& name); virtual void AddFeedVar(Variable* var, const std::string& name);
protected: protected:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册