From 3529a41266d67663ef477c71f0db12c52bb67884 Mon Sep 17 00:00:00 2001
From: YangYang <1525927685@qq.com>
Date: Sun, 19 Sep 2021 00:49:27 +0800
Subject: [PATCH] setup num_aug_repeats

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

diff --git a/train.py b/train.py
index 4be594e8..08b8e4c6 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=num_aug_splits,
+        num_aug_splits=args.num_aug_splits,
         interpolation=train_interpolation,
         mean=data_config['mean'],
         std=data_config['std'],