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