devit icon indicating copy to clipboard operation
devit copied to clipboard

How to inference in one shot mode?

Open LLsmile opened this issue 11 months ago • 6 comments

I modified demo\demo.py like

        config_file="configs/one-shot/split1_vitl.yaml",
        rpn_config_file="configs/RPN/mask_rcnn_R_50_FPN_1x.yaml",
        model_path="weights/trained/oneshot/vitl_0074999.split1.pth",

But it crashed with following messge:

RuntimeError: Error(s) in loading state_dict for OpenSetDetectorWithExamples:
	Missing key(s) in state_dict: "offline_backbone.fpn_lateral2.weight",

LLsmile avatar Feb 27 '24 05:02 LLsmile