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

update data_feed for add MultiSlotDataFeed

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