| _BASE_: "Base.yaml" | |
| MODEL: | |
| TRAIN_TASK: ["ObjectDet"] | |
| TEST_TASK: "ObjectDet" | |
| MASK_ON: True | |
| ROI_HEADS: | |
| SOFT_NMS_ENABLED: True | |
| BEAM_SIZE: 3 | |
| WEIGHTS: "detectron2://ImageNetPretrained/MAE/mae_pretrain_vit_base.pth" | |
| BACKBONE: | |
| NAME: build_vit_fpn_backbone | |
| VIT_LAYERS: 12 | |
| SOLVER: | |
| VIT_LAYER_DECAY_RATE: 0.7 | |
| DATASETS: | |
| TRAIN: ("GRiT_coco2017_train",) | |
| TEST: ("coco_2017_val",) | |
| DATALOADER: | |
| DATASET_BS: 2 | |
| OUTPUT_DIR: "./output/GRiT_B_ObjectDet" |