mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update autoaug and randaug configs

pull/913/head
mzr1996 2022-06-02 17:10:14 +08:00
parent 05e4c13343
commit 6563f5f448
14 changed files with 14 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
configs/_base_/datasets/imagenet_bs128_mbv3.py
View File

@ -1,5 +1,3 @@
_base_ = ['./pipelines/auto_aug.py']
# dataset settings # dataset settings
dataset_type = 'ImageNet' dataset_type = 'ImageNet'
preprocess_cfg = dict( preprocess_cfg = dict(
@ -19,7 +17,7 @@ train_pipeline = [
dict(type='RandomFlip', prob=0.5, direction='horizontal'), dict(type='RandomFlip', prob=0.5, direction='horizontal'),
dict( dict(
type='AutoAugment', type='AutoAugment',
policies={{_base_.auto_increasing_policies}}, policies='imagenet',
hparams=dict(pad_val=[round(x) for x in bgr_mean])), hparams=dict(pad_val=[round(x) for x in bgr_mean])),
dict( dict(
type='RandomErasing', type='RandomErasing',

4
configs/_base_/datasets/imagenet_bs128_poolformer_medium_224.py
View File

@ -1,5 +1,3 @@
_base_ = ['./pipelines/rand_aug.py']
# dataset settings # dataset settings
dataset_type = 'ImageNet' dataset_type = 'ImageNet'
preprocess_cfg = dict( preprocess_cfg = dict(
@ -23,7 +21,7 @@ train_pipeline = [
dict(type='RandomFlip', prob=0.5, direction='horizontal'), dict(type='RandomFlip', prob=0.5, direction='horizontal'),
dict( dict(
type='RandAugment', type='RandAugment',
policies={{_base_.rand_increasing_policies}}, policies='timm_increasing',
num_policies=2, num_policies=2,
total_level=10, total_level=10,
magnitude_level=9, magnitude_level=9,

4
configs/_base_/datasets/imagenet_bs128_poolformer_small_224.py
View File

@ -1,5 +1,3 @@
_base_ = ['./pipelines/rand_aug.py']
# dataset settings # dataset settings
dataset_type = 'ImageNet' dataset_type = 'ImageNet'
preprocess_cfg = dict( preprocess_cfg = dict(
@ -23,7 +21,7 @@ train_pipeline = [
dict(type='RandomFlip', prob=0.5, direction='horizontal'), dict(type='RandomFlip', prob=0.5, direction='horizontal'),
dict( dict(
type='RandAugment', type='RandAugment',
policies={{_base_.rand_increasing_policies}}, policies='timm_increasing',
num_policies=2, num_policies=2,
total_level=10, total_level=10,
magnitude_level=9, magnitude_level=9,

4
configs/_base_/datasets/imagenet_bs256_rsb_a12.py
View File

@ -1,5 +1,3 @@
_base_ = ['./pipelines/rand_aug.py']
# dataset settings # dataset settings
dataset_type = 'ImageNet' dataset_type = 'ImageNet'
preprocess_cfg = dict( preprocess_cfg = dict(
@ -23,7 +21,7 @@ train_pipeline = [
dict(type='RandomFlip', prob=0.5, direction='horizontal'), dict(type='RandomFlip', prob=0.5, direction='horizontal'),
dict( dict(
type='RandAugment', type='RandAugment',
policies={{_base_.rand_increasing_policies}}, policies='timm_increasing',
num_policies=2, num_policies=2,
total_level=10, total_level=10,
magnitude_level=7, magnitude_level=7,

4
configs/_base_/datasets/imagenet_bs256_rsb_a3.py
View File

@ -1,5 +1,3 @@
_base_ = ['./pipelines/rand_aug.py']
# dataset settings # dataset settings
dataset_type = 'ImageNet' dataset_type = 'ImageNet'
preprocess_cfg = dict( preprocess_cfg = dict(
@ -23,7 +21,7 @@ train_pipeline = [
dict(type='RandomFlip', prob=0.5, direction='horizontal'), dict(type='RandomFlip', prob=0.5, direction='horizontal'),
dict( dict(
type='RandAugment', type='RandAugment',
policies={{_base_.rand_increasing_policies}}, policies='timm_increasing',
num_policies=2, num_policies=2,
total_level=10, total_level=10,
magnitude_level=6, magnitude_level=6,

4
configs/_base_/datasets/imagenet_bs64_autoaug.py
View File

@ -1,5 +1,3 @@
_base_ = ['./pipelines/auto_aug.py']
# dataset settings # dataset settings
dataset_type = 'ImageNet' dataset_type = 'ImageNet'
preprocess_cfg = dict( preprocess_cfg = dict(
@ -19,7 +17,7 @@ train_pipeline = [
dict(type='RandomFlip', prob=0.5, direction='horizontal'), dict(type='RandomFlip', prob=0.5, direction='horizontal'),
dict( dict(
type='AutoAugment', type='AutoAugment',
policies={{_base_.auto_increasing_policies}}, policies='imagenet',
hparams=dict( hparams=dict(
pad_val=[round(x) for x in bgr_mean], interpolation='bicubic')), pad_val=[round(x) for x in bgr_mean], interpolation='bicubic')),
dict(type='PackClsInputs'), dict(type='PackClsInputs'),

4
configs/_base_/datasets/imagenet_bs64_convmixer_224.py
View File

@ -1,5 +1,3 @@
_base_ = ['./pipelines/rand_aug.py']
# dataset settings # dataset settings
dataset_type = 'ImageNet' dataset_type = 'ImageNet'
preprocess_cfg = dict( preprocess_cfg = dict(
@ -23,7 +21,7 @@ train_pipeline = [
dict(type='RandomFlip', prob=0.5, direction='horizontal'), dict(type='RandomFlip', prob=0.5, direction='horizontal'),
dict( dict(
type='RandAugment', type='RandAugment',
policies={{_base_.rand_increasing_policies}}, policies='timm_increasing',
num_policies=2, num_policies=2,
total_level=10, total_level=10,
magnitude_level=9, magnitude_level=9,

4
configs/_base_/datasets/imagenet_bs64_pil_resize_autoaug.py
View File

@ -1,5 +1,3 @@
_base_ = ['pipelines/auto_aug.py']
# dataset settings # dataset settings
dataset_type = 'ImageNet' dataset_type = 'ImageNet'
preprocess_cfg = dict( preprocess_cfg = dict(
@ -23,7 +21,7 @@ train_pipeline = [
dict(type='RandomFlip', prob=0.5, direction='horizontal'), dict(type='RandomFlip', prob=0.5, direction='horizontal'),
dict( dict(
type='AutoAugment', type='AutoAugment',
policies={{_base_.policy_imagenet}}, policies='imagenet',
hparams=dict( hparams=dict(
pad_val=[round(x) for x in bgr_mean], interpolation='bicubic')), pad_val=[round(x) for x in bgr_mean], interpolation='bicubic')),
dict(type='PackClsInputs'), dict(type='PackClsInputs'),

4
configs/_base_/datasets/imagenet_bs64_swin_224.py
View File

@ -1,5 +1,3 @@
_base_ = ['./pipelines/rand_aug.py']
# dataset settings # dataset settings
dataset_type = 'ImageNet' dataset_type = 'ImageNet'
preprocess_cfg = dict( preprocess_cfg = dict(
@ -23,7 +21,7 @@ train_pipeline = [
dict(type='RandomFlip', prob=0.5, direction='horizontal'), dict(type='RandomFlip', prob=0.5, direction='horizontal'),
dict( dict(
type='RandAugment', type='RandAugment',
policies={{_base_.rand_increasing_policies}}, policies='timm_increasing',
num_policies=2, num_policies=2,
total_level=10, total_level=10,
magnitude_level=9, magnitude_level=9,

4
configs/_base_/datasets/imagenet_bs64_t2t_224.py
View File

@ -1,5 +1,3 @@
_base_ = ['./pipelines/rand_aug.py']
# dataset settings # dataset settings
dataset_type = 'ImageNet' dataset_type = 'ImageNet'
preprocess_cfg = dict( preprocess_cfg = dict(
@ -23,7 +21,7 @@ train_pipeline = [
dict(type='RandomFlip', prob=0.5, direction='horizontal'), dict(type='RandomFlip', prob=0.5, direction='horizontal'),
dict( dict(
type='RandAugment', type='RandAugment',
policies={{_base_.rand_increasing_policies}}, policies='timm_increasing',
num_policies=2, num_policies=2,
total_level=10, total_level=10,
magnitude_level=9, magnitude_level=9,

2
configs/van/van-base_8xb128_in1k.py
View File

@ -26,7 +26,7 @@ train_pipeline = [
dict(type='RandomFlip', prob=0.5, direction='horizontal'), dict(type='RandomFlip', prob=0.5, direction='horizontal'),
dict( dict(
type='RandAugment', type='RandAugment',
policies={{_base_.rand_increasing_policies}}, policies='timm_increasing',
num_policies=2, num_policies=2,
total_level=10, total_level=10,
magnitude_level=9, magnitude_level=9,

2
configs/van/van-large_8xb128_in1k.py
View File

@ -26,7 +26,7 @@ train_pipeline = [
dict(type='RandomFlip', prob=0.5, direction='horizontal'), dict(type='RandomFlip', prob=0.5, direction='horizontal'),
dict( dict(
type='RandAugment', type='RandAugment',
policies={{_base_.rand_increasing_policies}}, policies='timm_increasing',
num_policies=2, num_policies=2,
total_level=10, total_level=10,
magnitude_level=9, magnitude_level=9,

2
configs/van/van-small_8xb128_in1k.py
View File

@ -26,7 +26,7 @@ train_pipeline = [
dict(type='RandomFlip', prob=0.5, direction='horizontal'), dict(type='RandomFlip', prob=0.5, direction='horizontal'),
dict( dict(
type='RandAugment', type='RandAugment',
policies={{_base_.rand_increasing_policies}}, policies='timm_increasing',
num_policies=2, num_policies=2,
total_level=10, total_level=10,
magnitude_level=9, magnitude_level=9,

2
configs/van/van-tiny_8xb128_in1k.py
View File

@ -26,7 +26,7 @@ train_pipeline = [
dict(type='RandomFlip', prob=0.5, direction='horizontal'), dict(type='RandomFlip', prob=0.5, direction='horizontal'),
dict( dict(
type='RandAugment', type='RandAugment',
policies={{_base_.rand_increasing_policies}}, policies='timm_increasing',
num_policies=2, num_policies=2,
total_level=10, total_level=10,
magnitude_level=9, magnitude_level=9,