Dense_Captioning_-_GRiT / detectron2 /configs /quick_schedules /keypoint_rcnn_R_50_FPN_instant_test.yaml
Vishakaraj's picture
Upload folder using huggingface_hub
c709b60
raw
history blame contribute delete
No virus
346 Bytes
_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