RMN icon indicating copy to clipboard operation
RMN copied to clipboard

a problem about sample.py

Open 1031051895 opened this issue 1 year ago • 0 comments

when I run sample.py line 102, in net.load_state_dict(torch.load(opt.model_pth_path)) RuntimeError: Error(s) in loading state_dict for CapModel: Unexpected key(s) in state_dict: "decoder.module_selection.loc_fc.weight", "decoder.module_selection.loc_fc.bias", "decoder.module_selection.rel_fc.weight", "decoder.module_selection.rel_fc.bias", "decoder.module_selection.func_fc.weight", "decoder.module_selection.func_fc.bias", "decoder.module_selection.module_attn.wh.weight", "decoder.module_selection.module_attn.wh.bias", "decoder.module_selection.module_attn.wv.weight", "decoder.module_selection.module_attn.wv.bias", "decoder.module_selection.module_attn.wa.weight". can you help me thank you very much

1031051895 avatar Apr 06 '23 06:04 1031051895