h2fa_rcnn_R_101_DC5_cliparttest.yaml

_BASE_: "../Base-RCNN-DilatedC5.yaml"
MODEL:
  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
  MASK_ON: False
  RESNETS:
    DEPTH: 101
    OUT_FEATURES: ["res2", "res5"]
  BACKBONE:
    FREEZE_AT: 0
  ROI_HEADS:
    NUM_CLASSES: 20
INPUT:
  MIN_SIZE_TRAIN: (480, 512, 544, 576, 608, 640, 672, 704, 736, 768, 800)
  MIN_SIZE_TEST: 800
TEST:
  EVAL_PERIOD: 5000
DATASETS:
  TRAIN: ('clipart1k_train',)
  TRAIN_SOURCE: ('voc_2007_trainval', 'voc_2012_trainval')
  TEST: ('clipart1k_test',)
SOLVER:
  STEPS: (16000, 21500)
  MAX_ITER: 24000
  WARMUP_ITERS: 400
  IMS_PER_BATCH: 4 # image number for each domain
  BASE_LR: 0.005 
OUTPUT_DIR: "./output"
DATALOADER:
  NUM_WORKERS: 4