loading the pre-trained model throwing an error
udayallu opened this issue · 1 comments
udayallu commented
loading the pre-trained model throwing an error
cfg = get_cfg()
#Use the final weights generated after successful training for inference
cfg.MODEL.WEIGHTS = "publaynet.pkl"
cfg.merge_from_file("faster_rcnn.yaml")
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.7
Non-existent config key: MODEL.TYPE
arjunpanicker commented
@udayallu This is a detectron model and you are trying to use it in detectron2 which is not possible.