标签：PyTorch

技能

模型推理部署——PyTorch篇

PyTorch中模型的保存和加载保存和加载模型参数（state_dict）（推荐）state_dice 是一个简单的 python 字典，映射了每一层的参数名称和数值。# savetorch.save(model.state_dict,PATH)#loadmodel = TheModelClass(*args, **kwargs)model.l……