RuntimeError: Error(s) in loading state_dict for SSD: Missing key(s) in解决办法

模型验证的时候运行出现RuntimeError: Error(s) in loading state_dict for SSD:Missing key(s) in state_dict: 解决办法|

net.load_state_dict(torch.load(args.trained_model))

改为

net.load_state_dict(torch.load(args.trained_model), False)

即可

主要是记录一下自己平时遇到的问题,和大家分享一下
如有侵犯,请联系我

点个赞支持一下吧

你可能感兴趣的:(RuntimeError: Error(s) in loading state_dict for SSD: Missing key(s) in解决办法)