Fix class aware sampling (#542)

a6ab91d5 · qingqing01 · GitHub · e040d0e6 · a6ab91d5
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppdet/data/reader.py ppdet/data/reader.py +1 -1

未找到文件。
--- a/ppdet/data/reader.py
+++ b/ppdet/data/reader.py
@@ -279,7 +279,7 @@ class Reader(object):
            self.indexes = np.random.choice(
                self._sample_num,
                self._sample_num,
-                replace=False,
+                replace=True,
                p=self.img_weights)
        if self._shuffle: