当前位置:网站首页>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的维度。
边栏推荐
猜你喜欢
Appium 自动化测试环境搭建
MySQL 表的增删查改
二、pytest+selenium+allure实现web ui自动化
iNFTnews | Metaverse brings new ideas for enterprise development
智能指针学习笔记
一、根据系统架构定位系统性能瓶颈
3 个开源项目,让你感受程序员的浪漫!
LeetCode_二叉树_中等_515.在每个树行中找最大值
字节一面:TCP 和 UDP 可以使用同一个端口吗?
The realization of the salary slip issuing function of WeChat public account + web background
随机推荐
D. Districts Connection
mysql 索引和 pgsql 索引 命名区别
laravel数据库: 查询构造器
Subject: Ordered Queue
The situation of the solution of the equation system and the correlation transformation of the vector group
bzoj1269 [AHOI2006]文本编辑器editor
MySQL 表的增删查改
差分约束做法
ERROR Failed to compile with 1 error
linux安装部署redis&配置远程连接
VISTA无人驾驶模拟器;FinRL量化金融深度强化学习库;『深度神经网络应用』电子书;CUDA/TensorRT案例集锦;前沿论文 | ShowMeAI资讯日报
D. Non-zero Segments
phar反序列化
用完华为云会议解决方案,我直接卸载了之前的会议软件【华为云至简致远】
[Paper Reading] RAL 2022: Receding Moving Object Segmentation in 3D LiDAR Data Using Sparse 4D Convolutions
pdf导出工具类
字节一面:TCP 和 UDP 可以使用同一个端口吗?
李沐:机器学习者进阶学习建议
一、根据系统架构定位系统性能瓶颈
Solve the inexplicable problem of MySQL violently - restart the service!