从.pts文件加载pytorch模型

从.pts文件加载pytorch模型,pytorch,conv-neural-network,Pytorch,Conv Neural Network,我想知道是否有人可以解释如何加载以.pts扩展名保存的pytorch模型并从文件中提取权重 更具体地说,我可以使用以下方式加载模型: model = torch.jit.load(model_path.pts,torch.device("cuda")) 属于以下类型: torch.jit._script.RecursiveScriptModule 我看了以下几点: 我可以用它向前传球,但我会尝试找出如何获得重量,这样我也可以对重量进行更新。这能解决您的问题吗@知识获取者

我想知道是否有人可以解释如何加载以.pts扩展名保存的pytorch模型并从文件中提取权重

更具体地说,我可以使用以下方式加载模型:

model = torch.jit.load(model_path.pts,torch.device("cuda"))
属于以下类型:

torch.jit._script.RecursiveScriptModule
我看了以下几点:


我可以用它向前传球,但我会尝试找出如何获得重量,这样我也可以对重量进行更新。

这能解决您的问题吗@知识获取者啊,是的,愚蠢的我lol模型。参数()对不起:)没关系,没问题,:D