Spaces:
Runtime error
Runtime error
Dense_Captioning_-_GRiT
/
detectron2
/configs
/quick_schedules
/keypoint_rcnn_R_50_FPN_instant_test.yaml
_BASE_: "../Base-RCNN-FPN.yaml" | |
MODEL: | |
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl" | |
KEYPOINT_ON: True | |
ROI_HEADS: | |
NUM_CLASSES: 1 | |
DATASETS: | |
TRAIN: ("keypoints_coco_2017_val_100",) | |
TEST: ("keypoints_coco_2017_val_100",) | |
SOLVER: | |
BASE_LR: 0.005 | |
STEPS: (30,) | |
MAX_ITER: 40 | |
IMS_PER_BATCH: 4 | |
DATALOADER: | |
NUM_WORKERS: 2 | |