fix config to match best model

01b21983 · Simon Geisler · GitHub · 1f8b5b27 · 01b21983
Unverified Commit 01b21983 authored Sep 03, 2021 by Simon Geisler Committed by GitHub Sep 03, 2021
Show whitespace changes
Inline Side-by-side

Showing with 0 additions and 2 deletions

official/vision/beta/projects/vit/configs/image_classification.py .../vision/beta/projects/vit/configs/image_classification.py +0 -2

No files found.
--- a/official/vision/beta/projects/vit/configs/image_classification.py
+++ b/official/vision/beta/projects/vit/configs/image_classification.py
@@ -110,8 +110,6 @@ def image_classification_imagenet_deit_pretrain() -> cfg.ExperimentConfig:
              input_path=os.path.join(IMAGENET_INPUT_PATH_BASE, 'train*'),
              is_training=True,
              global_batch_size=train_batch_size,
-              color_jitter=0.4,
-              random_erasing=common.RandomErasing(),
              aug_type=common.Augmentation(
                  type='randaug', randaug=common.RandAugment(
                      magnitude=9, exclude_ops=['Cutout'])),