当前位置:网站首页>pytorch常用语句
pytorch常用语句
2022-08-08 16:54:00 【泉伟】
1. 检测是否存在cuda
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") # PyTorch v0.4.0
model = CNN().to(device)
2. numpy与torch互转
# numpy 转为 torch
torch_data = torch.from_numpy(np_data)
# torch 转为numpy
tensor2array = torch_data.numpy()
3. 打印网络形状
需要提前安装torchsummary库,直接pip安装就是可以的。
from torchsummary import summary
if __name__ == "__main__":
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = CNN().to(device)
summary(model, (1, 3)) #(1,3)为模型输入
4. 保存加载权重
保存权重
# 建立一个字典,保存三个参数
state = {
‘net':model.state_dict(), 'optimizer':optimizer.state_dict(), 'epoch':epoch}
# 调用torch.save()
# dir表示保存文件的绝对路径+保存文件名,如'/home/qinying/Desktop/modelpara.pth'
torch.save(state, dir)
加载权重
checkpoint = torch.load(dir)
model.load_state_dict(checkpoint['net'])
optimizer.load_state_dict(checkpoint['optimizer'])
start_epoch = checkpoint['epoch'] + 1
5. 数据维度改变
5.1 维度压缩
使用函数torch.squeeze()对数据的维度进行压缩,去掉维数为1的的维度,比如是一行或者一列这种,一个一行三列(1,3)的数去掉第一个维数为一的维度之后就变成(3)行。squeeze(a)就是将a中所有为1的维度删掉。不为1的维度没有影响。a.squeeze(N) 就是去掉a中指定的维数为一的维度。
还有一种形式就是b=torch.squeeze(a,N) a中去掉指定的定的维数为一的维度。
5.2 维度扩充
torch.unsqueeze()函数主要是对数据维度进行扩充。给指定位置加上维数为一的维度,比如原本有个三行的数据(3),在0的位置加了一维就变成一行三列(1,3)。a.squeeze(N) 就是在a中指定位置N加上一个维数为1的维度。还有一种形式就是b=torch.squeeze(a,N) a就是在a中指定位置N加上一个维数为1的维度。
边栏推荐
猜你喜欢
用完华为云会议解决方案,我直接卸载了之前的会议软件【华为云至简致远】
看到这个应用上下线方式,不禁感叹:优雅,太优雅了!
VISTA无人驾驶模拟器;FinRL量化金融深度强化学习库;『深度神经网络应用』电子书;CUDA/TensorRT案例集锦;前沿论文 | ShowMeAI资讯日报
PNAS最新研究:81%解题率,神经网络 Codex 推开高等数学世界大门
linux安装部署redis&配置远程连接
【入门PCB】立创eda的学习
WPF StepBar
ctfshow七夕杯复现
The latest research from PNAS: 81% problem solving rate, neural network Codex opens the door to the world of advanced mathematics
元宇宙医疗或将改变医疗格局
随机推荐
leetcode:296.最佳的碰头地点
QCon 回顾 | Data Fabric:逻辑统一、物理分散
维尔薇vs千劫
4、S32K14X学习笔记:S32 Design Studio 新建和导入工程
它们不一样!透析【观察者模式】和【发布订阅模式】
淘宝API常用接口列表与申请方式
永续合约交易所系统开发逻辑详情
五、jmeter脚本的基本构成&断言
bzoj1251 序列终结者
LeetCode_Binary Tree_Medium_515. Find the maximum value in each tree row
PNAS最新研究:81%解题率,神经网络 Codex 推开高等数学世界大门
L2-011 玩转二叉树 (25 分) (二叉树)
ASP.NET Core依赖注入之旅:4.体验服务的注册和消费
The realization of the salary slip issuing function of WeChat public account + web background
C1. Pokémon Army (easy version)
IDEA2020安装教程
[深入研究4G/5G/6G专题-54]: L3信令控制-3-软件功能与流程的切分-CU-UP网元的信令
一、搭建django自动化平台(实现一键执行sql)
redis介绍&命令&性能相关&缓存穿透
Mysql都有那些最需要掌握的原理?