internimage_xl_22kto1k_384.yaml

DATA:
  IMG_SIZE: 384
  IMG_ON_MEMORY: True
AUG:
  MIXUP: 0.0
  CUTMIX: 0.0
  REPROB: 0.0
MODEL:
  TYPE: intern_image
  DROP_PATH_RATE: 0.2
  LABEL_SMOOTHING: 0.3
  INTERN_IMAGE:
    CORE_OP: 'DCNv3'
    DEPTHS: [5, 5, 24, 5]
    GROUPS: [12, 24, 48, 96]
    CHANNELS: 192
    LAYER_SCALE: 1e-5
    OFFSET_SCALE: 2.0
    MLP_RATIO: 4.0
    POST_NORM: True
  PRETRAINED: 'pretrained/internimage_xl_22k_192to384.pth'
TRAIN:
  EMA:
    ENABLE: true
    DECAY: 0.9999
  EPOCHS: 20
  WARMUP_EPOCHS: 2
  WEIGHT_DECAY: 0.05
  BASE_LR: 2e-05 # 512
  WARMUP_LR: .0
  MIN_LR: .0
  USE_CHECKPOINT: true
  OPTIMIZER:
    DCN_LR_MUL: 0.1
AMP_OPT_LEVEL: O0
EVAL_FREQ: 1