pytorch Missing key(s) in state_dict

加载模型的时候,加上一个关键字 strict


model.load_state_dict(pretrained_dict, strict=False)    # 多GPU训练,单GPU推理 strict=False

你可能感兴趣的:(算法学习,pytorch,深度学习,人工智能)