flatten_pvt_v2_b0.yaml

DATA:
  IMG_SIZE: 224
  BATCH_SIZE: 128

TRAIN:
  WEIGHT_DECAY: 0.05
  EPOCHS: 300
  WARMUP_EPOCHS: 5
  COOLDOWN_EPOCHS: 10
  BASE_LR: 5e-4
  WARMUP_LR: 1e-6
  MIN_LR: 1e-5
  CLIP_GRAD: None

MODEL:
  TYPE: flatten_pvt_v2_b0
  NAME: flatten_pvt_v2_b0
  DROP_PATH_RATE: 0.1
  LA:
    ATTN_TYPE: LLLL