From dd88037afa888fa6b2b5eca6d48241c1ceeea1d5 Mon Sep 17 00:00:00 2001 From: jerrywgz Date: Fri, 1 Feb 2019 15:21:53 +0800 Subject: [PATCH] Fix rcnn reader bug (#1725) * fix reader bug in RCNN --- fluid/PaddleCV/rcnn/reader.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/fluid/PaddleCV/rcnn/reader.py b/fluid/PaddleCV/rcnn/reader.py index 4a634720..bdfab3d3 100644 --- a/fluid/PaddleCV/rcnn/reader.py +++ b/fluid/PaddleCV/rcnn/reader.py @@ -127,6 +127,7 @@ def coco(mode, roidb_cur = 0 count = 0 batch_out = [] + device_num = total_batch_size / batch_size while True: roidb = roidb_perm[0] roidb_cur += 1 @@ -153,7 +154,7 @@ def coco(mode, else: if len(batch_out) == total_batch_size: batch_out = padding_minibatch(batch_out) - for i in range(total_batch_size / batch_size): + for i in range(device_num): sub_batch_out = [] for j in range(batch_size): sub_batch_out.append(batch_out[i * batch_size + @@ -162,7 +163,8 @@ def coco(mode, count += 1 sub_batch_out = [] batch_out = [] - if count >= cfg.max_iter: + iter_id = count // device_num + if iter_id >= cfg.max_iter: return elif mode == "test": batch_out = [] -- GitLab