Skip to content

GitLab

Unverified Commit 01b21983 authored by Simon Geisler's avatar Simon Geisler Committed by GitHub
Browse files

fix config to match best model

parent 1f8b5b27
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
official/vision/beta/projects/vit/configs/image_classification.py
View file @ 01b21983
...@@ -110,8 +110,6 @@ def image_classification_imagenet_deit_pretrain() -> cfg.ExperimentConfig: ...@@ -110,8 +110,6 @@ def image_classification_imagenet_deit_pretrain() -> cfg.ExperimentConfig:
input_path=os.path.join(IMAGENET_INPUT_PATH_BASE, 'train*'), input_path=os.path.join(IMAGENET_INPUT_PATH_BASE, 'train*'),
is_training=True, is_training=True,
global_batch_size=train_batch_size, global_batch_size=train_batch_size,
color_jitter=0.4,
random_erasing=common.RandomErasing(),
aug_type=common.Augmentation( aug_type=common.Augmentation(
type='randaug', randaug=common.RandAugment( type='randaug', randaug=common.RandAugment(
magnitude=9, exclude_ops=['Cutout'])), magnitude=9, exclude_ops=['Cutout'])),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment