File size: 485 Bytes
8fcd0f0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
_BASE_: "Base-RCNN-FPN.yml"
MODEL:
PIXEL_MEAN: [ 127.5, 127.5, 127.5 ]
PIXEL_STD: [ 127.5, 127.5, 127.5 ]
WEIGHTS: "https://layoutlm.blob.core.windows.net/dit/dit-pts/dit-base-224-p16-500k-62d53a.pth"
VIT:
NAME: "dit_base_patch16"
ROI_HEADS:
NAME: CascadeROIHeads
ROI_BOX_HEAD:
CLS_AGNOSTIC_BBOX_REG: True
RPN:
POST_NMS_TOPK_TRAIN: 2000
SOLVER:
WARMUP_ITERS: 1000
IMS_PER_BATCH: 16
MAX_ITER: 60000
CHECKPOINT_PERIOD: 2000
TEST:
EVAL_PERIOD: 2000 |