21 lines
486 B
YAML
21 lines
486 B
YAML
|
_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
|