onestage_grounding icon indicating copy to clipboard operation
onestage_grounding copied to clipboard

Dear author, during loading your trained model, what is the cause of the error? Looking forward to your reply!

Open lwj01 opened this issue 11 months ago • 0 comments

model_path = r'../saved_models/lstm_flickr_model.pth'

model = grounding_model.grounding_model(dataset='xxx')

state_dict = model.load_state_dict(torch.load(model_path)['state_dict'])

报错: Missing key(s) in state_dict: "visumodel.module_list.0.conv_0.weight", "visumodel.module_list.0.batch_norm_0.weight", "visumodel.module_list.0.batch_norm_0.bias", ... Unexpected key(s) in state_dict: "module.visumodel.module_list.0.conv_0.weight", "module.visumodel.module_list.0.batch_norm_0.weight",....

lwj01 avatar Feb 27 '24 16:02 lwj01