_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 |