提交 08479010 编写于 作者: K kolinwei 提交者: qingqing01

update image_classification train speed (#1597)

上级 f637491e
...@@ -306,7 +306,7 @@ def train(args): ...@@ -306,7 +306,7 @@ def train(args):
train_loss = np.array(train_info[0]).mean() train_loss = np.array(train_info[0]).mean()
train_acc1 = np.array(train_info[1]).mean() train_acc1 = np.array(train_info[1]).mean()
train_acc5 = np.array(train_info[2]).mean() train_acc5 = np.array(train_info[2]).mean()
train_speed = np.array(train_time).mean() / train_batch_size train_speed = np.array(train_time).mean() / (train_batch_size * device_num)
test_py_reader.start() test_py_reader.start()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册