diff --git a/fluid/PaddleCV/image_classification/train.py b/fluid/PaddleCV/image_classification/train.py index 8e53699ed09d1b8e2e55eb4b35e7a6bb41294720..28652286fbdef1c1d572c6150868042be30a2138 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()