From 97e15148a8331ec27a3c7c276b9fc2ee8ea13d57 Mon Sep 17 00:00:00 2001 From: minqiyang Date: Thu, 27 Sep 2018 13:32:29 +0800 Subject: [PATCH] Fix image_classification bug --- fluid/image_classification/reader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fluid/image_classification/reader.py b/fluid/image_classification/reader.py index 50be1cde..639b0b01 100644 --- a/fluid/image_classification/reader.py +++ b/fluid/image_classification/reader.py @@ -140,7 +140,7 @@ def _reader_creator(file_list, # distributed mode if the env var `PADDLE_TRAINING_ROLE` exits trainer_id = int(os.getenv("PADDLE_TRAINER_ID", "0")) trainer_count = int(os.getenv("PADDLE_TRAINERS", "1")) - per_node_lines = len(full_lines) / trainer_count + per_node_lines = len(full_lines) // trainer_count lines = full_lines[trainer_id * per_node_lines:(trainer_id + 1) * per_node_lines] print( -- GitLab