Pytorch 读取t7文件

Pytorch 1.0以上可以使用:

c 复制代码
import torchfile

th_path = r"./path/xx.t7"
data = torchfile.load(th_path)

print(data.shape)

若data的尺寸为0,则将torch版本降为0.4.1,并使用以下函数:

c 复制代码
from torch.utils.serialization import load_lua

th_path = r"./path/xx.t7"
data = load_lua(th_path).numpy()

print(data.shape)

注意:

若是在Windows的系统中读取t7文件,一定要记得要用long_size=8

c 复制代码
data = torchfile.load(th_path,long_size=8)
或
data = load_lua(th_path,long_size=8).numpy()
相关推荐
DevinLGT19 分钟前
6Pin Type-C Pin脚定义:【图文讲解】
人工智能·单片机·嵌入式硬件
宋一诺3323 分钟前
机器学习—高级优化方法
人工智能·机器学习
feifeikon25 分钟前
Python Day5 进阶语法(列表表达式/三元/断言/with-as/异常捕获/字符串方法/lambda函数
开发语言·python
龙的爹233336 分钟前
论文 | The Capacity for Moral Self-Correction in LargeLanguage Models
人工智能·深度学习·机器学习·语言模型·自然语言处理·prompt
Mr.简锋39 分钟前
opencv视频读写
人工智能·opencv·音视频
Baihai_IDP39 分钟前
「混合专家模型」可视化指南:A Visual Guide to MoE
人工智能·llm·aigc
杰仔正在努力1 小时前
python成长技能之枚举类
开发语言·python
Eiceblue1 小时前
通过Python 调整Excel行高、列宽
开发语言·vscode·python·pycharm·excel
寰宇视讯1 小时前
“津彩嘉年,洽通天下” 2024中国天津投资贸易洽谈会火热启动 首届津彩生活嘉年华重磅来袭!
大数据·人工智能·生活
Jam-Young1 小时前
Python中的面向对象编程,类,对象,封装,继承,多态
开发语言·python