pytorch加载.pth格式文件

pytorch怎么读取.pth文件,以及怎么将torch的.pth文件转化为bin文件进行保存。

import torch
import numpy as np

path_in="in.pth"
path_out="out.pth"

#读取文件
#若安装的pytorch是带Cuda的版本,cuda不能用,则需要指定为cpu运行
data_in = torch.load(path_in, 'cpu')
data_out = torch.load(path_out, 'cpu')
#将torch的数据格式转化为numpy的数据格式
in_data = data_in.numpy().astype(np.uint8)
data_out = data_out.numpy().astype(np.uint8)
#保存文件
in_data.tofile("in.bin")
data_out.tofile("out.bin")

你可能感兴趣的:(python,numpy,python)