Res-VMamba / config.json
ms57rd's picture
fix class number
ac94ddd verified
AMP_ENABLE: true
AMP_OPT_LEVEL: ''
AUG:
AUTO_AUGMENT: rand-m9-mstd0.5-inc1
COLOR_JITTER: 0.4
CUTMIX: 1.0
CUTMIX_MINMAX: null
MIXUP: 0.8
MIXUP_MODE: batch
MIXUP_PROB: 1.0
MIXUP_SWITCH_PROB: 0.5
RECOUNT: 1
REMODE: pixel
REPROB: 0.25
BASE:
- ''
DATA:
BATCH_SIZE: 128
CACHE_MODE: part
DATASET: imagenet
DATA_PATH: /home/public_3T/food_data/CNFOOD-241
IMG_SIZE: 224
INTERPOLATION: bicubic
MASK_PATCH_SIZE: 32
MASK_RATIO: 0.6
NUM_WORKERS: 8
PIN_MEMORY: true
ZIP_MODE: false
ENABLE_AMP: false
EVAL_MODE: false
FUSED_LAYERNORM: false
FUSED_WINDOW_PROCESS: false
LOCAL_RANK: 0
MODEL:
DROP_PATH_RATE: 0.3
DROP_RATE: 0.0
LABEL_SMOOTHING: 0.1
MMCKPT: false
NAME: vssm_small
NUM_CLASSES: 1000
PRETRAINED: ./res_vmamba_cnf241_result_2/vssm_small/default/ckpt_epoch_12.pth
RESUME: ''
TYPE: vssm
VSSM:
DEPTHS:
- 2
- 2
- 27
- 2
DOWNSAMPLE: v1
DT_RANK: auto
D_STATE: 16
EMBED_DIM: 96
IN_CHANS: 3
MLP_RATIO: 0.0
PATCH_NORM: true
PATCH_SIZE: 4
SHARED_SSM: false
SOFTMAX: false
SSM_RATIO: 2.0
OUTPUT: ./res_vmamba_cnf241_result_best/vssm_small/default
PRINT_FREQ: 10
SAVE_FREQ: 1
SEED: 0
TAG: default
TEST:
CROP: true
SEQUENTIAL: false
SHUFFLE: false
THROUGHPUT_MODE: false
TRAIN:
ACCUMULATION_STEPS: 1
AUTO_RESUME: true
BASE_LR: 0.000125
CLIP_GRAD: 5.0
EPOCHS: 300
LAYER_DECAY: 1.0
LR_SCHEDULER:
DECAY_EPOCHS: 30
DECAY_RATE: 0.1
GAMMA: 0.1
MULTISTEPS: []
NAME: cosine
WARMUP_PREFIX: true
MIN_LR: 1.25e-06
MOE:
SAVE_MASTER: false
OPTIMIZER:
BETAS:
- 0.9
- 0.999
EPS: 1.0e-08
MOMENTUM: 0.9
NAME: adamw
START_EPOCH: 0
USE_CHECKPOINT: false
WARMUP_EPOCHS: 20
WARMUP_LR: 1.25e-07
WEIGHT_DECAY: 0.05