pytorch加载.pth格式文件

最新推荐文章于 2024-07-10 11:08:56 发布

一杯盐水

最新推荐文章于 2024-07-10 11:08:56 发布

阅读量2.6k

点赞数 2

分类专栏： python 文章标签： numpy python

本文链接：https://blog.csdn.net/CFH1021/article/details/105200792

版权

python 专栏收录该内容

8 篇文章 2 订阅

订阅专栏

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")