Trans2Seg_medium_512x512_80k_trans10kv2_bs_16.yaml

DATA:
    DATASET: "Trans10kV2"
    BATCH_SIZE: 16
    BATCH_SIZE_VAL: 1
    DATA_PATH: 'E:/Trans10K_cls12'
    CROP_SIZE: (512, 512)
    NUM_CLASSES: 12
TRAIN:
    BASE_LR: 0.0001
    ITERS: 80000
    LR_SCHEDULER:
        NAME: "PolynomialDecay"
    OPTIMIZER:
        NAME: 'ADAM'
VAL:
    MULTI_SCALES_VAL: False
    IMAGE_BASE_SIZE: 512
    CROP_SIZE: [512, 512]
MODEL:
    NAME: "Trans2Seg"
    ENCODER:
        TYPE: "resnet50c"
        MULTI_GRID: 
        MULTI_DILATION: 
    TRANS2SEG:
        EMBED_DIM: 256
        DEPTH: 4
        NUM_HEADS: 8
        MLP_RATIO: 3.
        HID_DIM: 64
SAVE_DIR: "./output/trans10kv2/Trans2Seg_medium_512x512_80k_trans10kv2_bs_16"