|
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 |
|
|