From 9de8192dee1edbd8e087dc85cb8a56cedc8060cb Mon Sep 17 00:00:00 2001
From: YangYang <1525927685@qq.com>
Date: Sun, 19 Sep 2021 00:52:54 +0800
Subject: [PATCH] remove args.aug_repeats

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 08b8e4c6..4be594e8 100755
--- a/train.py
+++ b/train.py
@@ -557,7 +557,7 @@ def main():
         vflip=args.vflip,
         color_jitter=args.color_jitter,
         auto_augment=args.aa,
-        num_aug_splits=args.num_aug_splits,
+        num_aug_splits=num_aug_splits,
         interpolation=train_interpolation,
         mean=data_config['mean'],
         std=data_config['std'],