加载模型出错
RuntimeError: Error(s) in loading state_dict for DataParallel:
Unexpected key(s) in state_dict: “module.sa_conv.0.weight”.
解决方法:logger.info("=> loading weight ‘{}’".format(args.weight))
checkpoint = torch.load(args.weight)
model.load_state_dict(checkpoint[‘state_dict’],False)
logger.info("=> loaded weight ‘{}’".format(args.weight))
加个False,不写默认为True