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()
相关推荐
碣石潇湘无限路12 分钟前
【AI篇】当Transformer模型开始学习《孙子兵法》
人工智能·学习
看到我,请让我去学习23 分钟前
OpenCV开发-初始概念
人工智能·opencv·计算机视觉
汀沿河23 分钟前
8.1 prefix Tunning与Prompt Tunning模型微调方法
linux·运维·服务器·人工智能
陈敬雷-充电了么-CEO兼CTO33 分钟前
大模型技术原理 - 基于Transformer的预训练语言模型
人工智能·深度学习·语言模型·自然语言处理·chatgpt·aigc·transformer
学术 学术 Fun39 分钟前
✨ OpenAudio S1:影视级文本转语音与语音克隆Mac整合包
人工智能·语音识别
萧鼎1 小时前
深度探索 Py2neo:用 Python 玩转图数据库 Neo4j
数据库·python·neo4j
华子w9089258591 小时前
基于 Python Django 和 Spark 的电力能耗数据分析系统设计与实现7000字论文实现
python·spark·django
风铃喵游1 小时前
让大模型调用MCP服务变得超级简单
前端·人工智能
Rockson2 小时前
使用Ruby接入实时行情API教程
javascript·python
booooooty2 小时前
基于Spring AI Alibaba的多智能体RAG应用
java·人工智能·spring·多智能体·rag·spring ai·ai alibaba