From 9b656c3da238b81fa947191c1bbdc4fc4989d045 Mon Sep 17 00:00:00 2001
From: rrjia <360442573@qq.com>
Date: Thu, 28 Jul 2022 17:48:27 +0800
Subject: [PATCH] Update build.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

训练过程中，内存一直增长，到后期会把整个服务器的内存都占完，初步分析将上述两个增强关闭后，内存使用会正常，具体是因为哪个方法，因为哪个实现有问题，暂时还没有时间分析
---
 fastreid/data/transforms/build.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/fastreid/data/transforms/build.py b/fastreid/data/transforms/build.py
index 5ca06a75..6f0495d2 100644
--- a/fastreid/data/transforms/build.py
+++ b/fastreid/data/transforms/build.py
@@ -80,13 +80,13 @@ def build_transforms(cfg, is_train=True):
         if do_affine:
             res.append(T.RandomAffine(degrees=10, translate=None, scale=[0.9, 1.1], shear=0.1, resample=False,
                                       fillcolor=0))
-        if do_augmix:
-            res.append(AugMix(prob=augmix_prob))
+#         if do_augmix:
+#             res.append(AugMix(prob=augmix_prob))
         res.append(ToTensor())
         if do_rea:
             res.append(T.RandomErasing(p=rea_prob, value=rea_value))
-        if do_rpt:
-            res.append(RandomPatch(prob_happen=rpt_prob))
+#         if do_rpt:
+#             res.append(RandomPatch(prob_happen=rpt_prob))
     else:
         size_test = cfg.INPUT.SIZE_TEST
         do_crop = cfg.INPUT.CROP.ENABLED