force to not cache ram when stop strong augmentation. (meituan#1003)

ZHEQIUSHUI · Jan 9, 2024 · 87dd3d3 · 87dd3d3
1 parent e9656c3
commit 87dd3d3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/yolov6/core/engine.py b/yolov6/core/engine.py
@@ -326,6 +326,7 @@ def before_epoch(self):
         if self.epoch == self.max_epoch - self.args.stop_aug_last_n_epoch:
             self.cfg.data_aug.mosaic = 0.0
             self.cfg.data_aug.mixup = 0.0
+            self.args.cache_ram = False # disable cache ram when stop strong augmentation.
             self.train_loader, self.val_loader = self.get_data_loader(self.args, self.cfg, self.data_dict)
         self.model.train()
         if self.rank != -1: