From 1634787b6bab5ce1482b5ebcac3a016b3fc60b1c Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Mon, 31 Aug 2020 15:02:37 +0800 Subject: [PATCH] fix mixup & cutmix in dygraph (#1321) --- ppdet/data/reader.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ppdet/data/reader.py b/ppdet/data/reader.py index 5cd9f3971..ae670c27a 100644 --- a/ppdet/data/reader.py +++ b/ppdet/data/reader.py @@ -25,6 +25,7 @@ import numpy as np import logging from ppdet.core.workspace import register, serializable +from paddle.fluid.dygraph.parallel import ParallelEnv from .parallel_map import ParallelMap from .transform.batch_operators import Gt2YoloTarget @@ -244,8 +245,8 @@ class Reader(object): self._drop_empty = drop_empty # sampling - self._mixup_epoch = mixup_epoch - self._cutmix_epoch = cutmix_epoch + self._mixup_epoch = mixup_epoch // ParallelEnv().nranks + self._cutmix_epoch = cutmix_epoch // ParallelEnv().nranks self._class_aware_sampling = class_aware_sampling self._load_img = False -- GitLab