pytorch的技巧记录

最新推荐文章于 2024-07-14 17:45:00 发布
最新推荐文章于 2024-07-14 17:45:00 发布
阅读量189 收藏
点赞数
分类专栏： pytorch updating 文章标签： pytorch 深度学习 python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/random_repick/article/details/126002348
版权

PyTorch 张量 nonzero 条件筛选 大于1
关键词由CSDN通过智能技术生成
  • tensor大于1的位置，也可以是等于小于0之类的
a=torch.arange(24).view(2,3,4)
b=torch.nonzero(a>1, as_tuple=True) #默认False，读者可以试试
刀么克瑟拉莫
关注
专栏目录
Pytorch入门教程
专注大数据与人工智能技术分享，欢迎私信加群互相学习！
03-08 3万+
Pytorch学习总结:1.张量Tensor 张量是一种特殊的数据结构，与数组和矩阵非常相似。在PyTorch中，我们使用张量对模型的输入和输出以及模型的参数进行编码。 张量类似于NumPy的ndarray，除了张量可以在 GPU 或其他硬件加速器上运行。事实上，张量和NumPy数组通常可以共享相同的底层内存，从而无需复制数据。...
PyTorch技巧记录
xiweijie的博客
10-12 81
这里写目录标题PyTorch技巧记录PyTorch中的contiguous使用 PyTorch技巧记录 PyTorch中的contiguous使用 PyTorch中的contiguous使用
PyTorch小技巧记录
weixin_41377325的博客
12-01 332
记录一下常用的网络操作小技巧 有三个相同结构的网络模型, 并且参数名也相同，将其中两个模型的参数取平均作为第三个模型的参数 。 只融合某一层。 # model # alice_model # bob_model # 网络参数 ['conv1.weight', 'conv1.bias', 'conv2.weight', 'conv2.bias', 'fc1.weight', 'fc1.bias', 'fc2.weight', 'fc2.bias'] # 只融合一层 with torch.no_
PyTorch学习记录——PyTorch进阶训练技巧
maximejia的博客
07-20 1022
Pytorch学习记录，希望能够继续熟悉Pytorch程序编写。
Pytorch常用技巧记录
tongshen的博客
12-02 781
Pytorch常用技巧记录 目录 文章目录Pytorch常用技巧记录1、指定GPU编号2、查看模型每层输出详情3、梯度裁剪（Gradient Clipping）4、扩展单张图片维度5、独热编码6、防止验证模型时爆显存7、学习率衰减8、冻结某些层的参数9、对不同层使用不同学习率10、模型相关操作11、Pytorch内置one_hot函数转载 1、指定GPU编号 设置当前使用的GPU设备仅为0号设备...
PyTorch 常用的技巧 36个
最新发布
刘念卿的博客
07-14 458
数据处理方面，使用Dataset和DataLoader处理数据，应用transforms进行预处理和增强，并加速数据加载。模型构建方面，使用torch.nn.init初始化权重，保存和加载模型状态。训练过程中，采用学习率调度和混合精度训练。调试时，进行张量维度检查、模型结构打印和显存监控。性能优化包括多GPU训练和异步数据加载。高效推理方面，应用梯度累积和梯度裁剪。
Pytorch进阶训练技巧
UIBE_day_day_up的博客
11-22 350
记录了包括自定义损失函数、动态调整学习率、使用预训练模型、半精度训练等进阶技巧
PyTorch进阶训练技巧
Alexa_的博客
07-20 669
pytorch高阶训练技巧
Pytorch使用和技巧
Amazing_Fly的博客
08-02 1089
pytorch学习笔记
Pytorch实用技巧
热门推荐
NEWPLAN的专栏
11-26 1万+
Pytorch实用技巧 手动搭建梯度 构造一个这样的函数：y=(x+1)*3 import torch.nn as nn import torch as th def show_tensor(tensor, name="None"): print("\n***********%s*************" % name) print("Tensor: ", tensor) ...
Pytorch基础学习（第七章-Pytorch训练技巧）
Billie使劲学的博客
04-04 3672
课程一览表： 一、模型保存与加载 二、模型finetune 三、GPU的使用 四、Pytorch常见报错
【实习记录】pytorch学习（持续更新）
Saedi_0101的博客
08-25 594
pytorch学习心得
PyTorch学习笔记-7.PyTorch训练技巧
ruoqi23的博客
08-03 2536
7.PyTorch训练技巧 7.1.模型保存与加载 保存： torch.save 主要参数： • obj：对象 • f：输出路径 保存分为两种模式： 1:保存整个Module，即保存了整个模型的框架和参数 torch.save(net, path) 2:保存模型参数，即只保存模型的参数，下次使用时需要自己重新构建框架 state_dict = net.state_dict() torch.save(state_dict , path) 加载： 2. torch.load 主要参数 • f：..
ERNIE转化为pytorch可用模型记录
初级菜鸟
12-13 3066
据说ERNIE 在中文NLP 任务中比Bert更为优秀，看论文感觉是在bert基础上做了一些训练的技巧. https://github.com/nghuyong/ERNIE-Pytorch转化模型代码项目 测试代码附上可直接执行的： #!/usr/bin/env python # encoding: utf-8 import torch from pytorch_transforme...
grid_map教程（初见）
random_repick的专栏
09-21 5786
一、安装方法1：本人用的方法2sudo apt-get install ros-kinetic-grid-map方法2：需要先安装ROScd catkin_ws/src git clone https://github.com/ethz-asl/grid_map.git cd ../ catkin_make -DCMAKE_BUILD_TYPE=Release二、功能介绍1.octomap转化gri
Ubuntu常用命令(统计个数，计算容量，sed，sort)
random_repick的专栏
06-29 3518
每搜索一个命令就添加，更新中…… 1.统计文件个数： ll | wc -l 2.计算当前文件的容量： du -h --max-depth=0 . 3.计算~/Downloads子文件的容量： du -h --max-depth=1 ~/Downloads 4.查看已安装软件包（如ssh）： dpkg -l | grep ssh 5.查找动态库：...
小白都能看懂的rk3399 pro使用
random_repick的专栏
05-18 3516
前言 在ubuntu上给rk3399 pro烧录debian 1. 准备 从这里下载debian固件 从这里下载烧录工具 2. 烧录 1、Type-C线连接主机端的USB接口和TB-RK3399Pro开发板的Type-C接口。 2、长按TB-RK3399Pro开发板上recovery按键后重启机器，进入Loader模式。 3、解压固件，将固件拷贝到linuxTool-v1.0/images目录下 4、执行sudo ./flash.py -l all，如何失败执行sudo python flash.py
ubuntu16.04+GTX1660深度学习相关配置
random_repick的专栏
11-06 2610
1.前情提要 本文使用cuda10.1，cuda8不行，cuda9没试过 cudnn要使用显卡以后发布的版本 考虑到官网总是挂，可以在此下载cuda10.1，cudnn cudnn的developer版本安装失败，runtime版本缺少头文件，所以安装runtime版本，使用developer版本是头文件，注意头文件的命名 2.编译caffe遇到的问题 1、Check failed: err...
python中让人惊叹的命令
random_repick的专栏
01-10 2577
1.安装pytorch时，遇到问题：Bad CRC-32 for file 'torch/lib/libtorch_cuda.so'，使用参数-no-cache即可解决，即pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html --no-cache ...
PyTorch中文教程全方位解析
- CNAME：可能是一个指向PyTorch官方文档的别名记录文件，用于配置自定义域名。 - index.html：网站的主页文件，包含PyTorch教程的入口和导航信息。 - 404.html：自定义的404错误页面，当用户访问不存在的链接时...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

刀么克瑟拉莫

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值