R_101_X_32x8d_NAS_densebox.yaml

MODEL:
  META_ARCHITECTURE: "SingleStageDetector"
  WEIGHT: "models/R-101X-NAS.pth"
  RPN_ONLY: True
  DENSEBOX_ON: True
  BACKBONE:
    CONV_BODY: "R-101-NAS"
    OUT_CHANNELS: 256
    ENCODER_OUT_CHANNELS: (256, 512, 1024, 2048)
  RESNETS:
    STRIDE_IN_1X1: False
    BACKBONE_OUT_CHANNELS: 256
    NUM_GROUPS: 32
    WIDTH_PER_GROUP: 8
  RETINANET:
    SCALES_PER_OCTAVE: 3
    STRADDLE_THRESH: -1
    NUM_CLASSES: 81
    NMS_TH: 0.6
    USE_C5: False
DATASETS:
  TRAIN: ("coco_2017_train",)
  TEST: ("coco_2017_val",)
INPUT:
  MIN_SIZE_RANGE_TRAIN: (640, 800)
  MAX_SIZE_TRAIN: 1333
  MIN_SIZE_TEST: 800
  MAX_SIZE_TEST: 1333
DATALOADER:
  SIZE_DIVISIBILITY: 32
  NUM_WORKERS: 2
SEARCH:
  NAS_DECODER_ON: True
  DECODER:
    VERSION: 2
    CONFIG: [[[3, 3, 4, 4, 1]], [[3, 3, 4, 4, 1]], [[4, 3, 4, 3, 1]], [[1, 3, 4, 3, 1]], [[5, 3, 4, 4, 1]], [[4, 7, 4, 4, 1]], [[7, 2, 4, 4, 1]]]
    AGG_SIZE: 256
    NUM_CELLS: 7
SOLVER:
  BASE_LR: 0.01
  WEIGHT_DECAY: 0.0001
  STEPS: (120000, 160000)
  MAX_ITER: 180000
  IMS_PER_BATCH: 16
  NUM_GPUS: 4
  WARMUP_METHOD: "constant"
TEST:
  IMS_PER_BATCH: 8
OUTPUT_DIR: 'output/coco/r101X-NAS'