pretrain.yaml

_BASE_: "../Base_det.yaml"

MODEL:
  WEIGHTS: "detectron2://ImageNetPretrained/torchvision/R-50.pkl"
  TRANSFORMER:
    VOC_SIZE: 5462
  
DATASETS:
  TRAIN: ("chnsyn_train", "rects_train", "rects_val", "lsvt_train", "art_train",)
  TEST: ("totaltext_test",)

SOLVER:
  IMS_PER_BATCH: 8
  BASE_LR: 1e-4
  LR_BACKBONE: 1e-5
  WARMUP_ITERS: 0
  STEPS: (300000,)
  MAX_ITER: 400000
  CHECKPOINT_PERIOD: 200000

TEST:
  EVAL_PERIOD: 100000000

OUTPUT_DIR: "output/R50/rects/pretrain"