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'],