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