提交 4caaa288 编写于 作者: D dengkaipeng

unset feed_list

上级 03bd1ec6
......@@ -91,8 +91,6 @@ def main():
loader = DataLoader(dataset,
batch_sampler=batch_sampler,
places=device,
feed_list=[i.forward() for i in inputs + labels] \
if not FLAGS.dynamic else None,
num_workers=FLAGS.num_workers,
return_list=True,
collate_fn=train_collate_fn)
......@@ -116,8 +114,6 @@ def main():
loader = DataLoader(dataset,
batch_sampler=batch_sampler,
places=device,
feed_list=[i.forward() for i in inputs + labels] \
if not FLAGS.dynamic else None,
num_workers=FLAGS.num_workers,
return_list=True,
collate_fn=eval_collate_fn)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册