search

ibm-aur-nlp / PubLayNet

Other
900 stars 165 forks source link

loading the pre-trained model throwing an error #39

Open udayallu opened 3 years ago

udayallu commented 3 years ago

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 3 years ago

@udayallu This is a detectron model and you are trying to use it in detectron2 which is not possible.