From 0847901058347fb7a3fb5eede0ee39317583e895 Mon Sep 17 00:00:00 2001 From: kolinwei <331911734@qq.com> Date: Fri, 4 Jan 2019 13:32:26 +0800 Subject: [PATCH] update image_classification train speed (#1597) --- fluid/PaddleCV/image_classification/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fluid/PaddleCV/image_classification/train.py b/fluid/PaddleCV/image_classification/train.py index 8e53699e..28652286 100644 --- a/fluid/PaddleCV/image_classification/train.py +++ b/fluid/PaddleCV/image_classification/train.py @@ -306,7 +306,7 @@ def train(args): train_loss = np.array(train_info[0]).mean() train_acc1 = np.array(train_info[1]).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() -- GitLab