setup num_aug_repeats

4 years ago · 3529a41266
parent 83a60f55fc
commit 3529a41266
1 changed files with 1 additions and 1 deletions
--- 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'],