Skip to content

M
models

PaddlePaddle / models
大约 1 年 前同步成功

通知 222
Star 6828
Fork 2962
M
models
提交 1f64e6a5 编写于 作者: Y Yibing Liu
浏览文件
操作

Replace feed_dict by feed

上级 991e6c25
隐藏空白更改
内联 并排
Showing with 2 addition and 4 deletion
fluid/image_classification/train.py
浏览文件 @ 1f64e6a5
...@@ -220,8 +220,7 @@ def train_parallel_exe(args, ...@@ -220,8 +220,7 @@ def train_parallel_exe(args,
test_info = [[], [], []] test_info = [[], [], []]
for batch_id, data in enumerate(train_reader()): for batch_id, data in enumerate(train_reader()):
t1 = time.time() t1 = time.time()
loss, acc1, acc5 = train_exe.run(fetch_list, loss, acc1, acc5 = train_exe.run(fetch_list, feed=feeder.feed(data))
feed_dict=feeder.feed(data))
t2 = time.time() t2 = time.time()
period = t2 - t1 period = t2 - t1
loss = np.mean(np.array(loss)) loss = np.mean(np.array(loss))
...@@ -243,8 +242,7 @@ def train_parallel_exe(args, ...@@ -243,8 +242,7 @@ def train_parallel_exe(args,
train_acc5 = np.array(train_info[2]).mean() train_acc5 = np.array(train_info[2]).mean()
for data in test_reader(): for data in test_reader():
t1 = time.time() t1 = time.time()
loss, acc1, acc5 = test_exe.run(fetch_list, loss, acc1, acc5 = test_exe.run(fetch_list, feed=feeder.feed(data))
feed_dict=feeder.feed(data))
t2 = time.time() t2 = time.time()
period = t2 - t1 period = t2 - t1
loss = np.mean(np.array(loss)) loss = np.mean(np.array(loss))
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或