提交 2178f662 编写于 作者: F frankwhzhang

fix model.py

上级 efcd1c08
...@@ -142,6 +142,7 @@ class Model(object): ...@@ -142,6 +142,7 @@ class Model(object):
def train_net(self): def train_net(self):
input_data = self.input_data(is_infer=False) input_data = self.input_data(is_infer=False)
self._data_var = input_data self._data_var = input_data
# if use dataset _data_loader not used
self._data_loader = fluid.io.DataLoader.from_generator( self._data_loader = fluid.io.DataLoader.from_generator(
feed_list=self._data_var, feed_list=self._data_var,
capacity=64, capacity=64,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册