python如何打开pth，Python如何打开.pth文件

Python中加载.pth文件的方法

在Python中，可以使用torch.load()函数来加载.pth文件，下面是一个简单的示例：

import torch
加载.pth文件
model = torch.load('model.pth')
打印模型信息
print(model)

在上面的代码中，model.pth是要加载的模型文件的路径。torch.load()函数会读取该文件，并返回模型对象，我们可以使用该模型对象进行后续的工作，如使用模型进行预测或训练等。

需要注意的是，pth文件中存储的是多个模型，那么需要使用map_location参数来指定要加载的模型的位置。

import torch
加载第一个模型
model1 = torch.load('models.pth', map_location=0)
加载第二个模型
model2 = torch.load('models.pth', map_location=1)

在上面的代码中，models.pth文件中存储了两个模型，我们通过map_location参数来指定要加载的模型的位置，0表示第一个模型，1表示第二个模型。

如果需要在不同的设备之间迁移模型，可以使用torch.save()函数将模型保存到磁盘上，然后使用torch.load()函数在另一个设备上加载该模型。

import torch
在设备1上保存模型
device1_model = torch.save(model, 'model.pth')
在设备2上加载模型
device2_model = torch.load('model.pth', map_location='cpu:0')

在上面的代码中，model是要保存的模型对象，device1_model是保存模型后返回的对象，device2_model是在设备2上加载模型后返回的对象，注意，我们需要使用map_location参数来指定要在哪个设备上加载模型。