args.weights=“xxxxxxxx.pth”
weights_dict = torch.load(args.weights) #模型加载到电脑中
model.load_state_dict(weights_dict, strict=False)
args.weights=“xxxxxxxx.pth”
weights_dict = torch.load(args.weights) #模型加载到电脑中
model.load_state_dict(weights_dict, strict=False)