网站首页 站内搜索

搜索结果

查询Tags标签: torch,共有 290条记录
  • 张量的索引、分片、合并以及维度调整

    文章目录 张量的符号索引一维张量索引二维张量索引 tensor.view()方法张量的分片函数分块:chunk函数拆分:split函数 张量的合并操作拼接函数:cat堆叠函数:stack张量的符号索引 张量也是有序序列,我们可以根据每个元素在系统内的顺序“编号”,来找出特定的元素,也就…

    2021/6/15 10:51:30 人评论 次浏览
  • 图神经网络学习-task01简单图论与环境配置

    一、图论基础知识 连通图二、拉普拉斯矩阵 拉普拉斯矩阵(Laplacian matrix) 也叫做导纳矩阵、基尔霍夫矩阵或离散拉普拉斯算子,主要应用在图论中,作为一个图的矩阵表示。 定义图 度矩阵 邻接矩阵 拉普拉斯矩阵 三、配置 极客云:https://www.jikecloud.net/l…

    2021/6/15 10:24:03 人评论 次浏览
  • 语义分割学习笔记2——Jetson Xavier安装pytorch

    语义分割学习笔记2——Jetson Xavier安装pytorch 一、whl文件安装pytorch二、安装torchvision三、测试是否安装成功运行语义分割的代码,需要先安装pytorch,下面是我在Xavier上安装语义分割的步骤。 我的环境说明: 名称版本硬件:Nvidia Jetson Xavier AGXCUDA版本:10.…

    2021/6/7 10:26:49 人评论 次浏览
  • 【图像分类】——来来来,干了这碗EfficientNet实战(Pytorch)

    目录 摘要 新建项目 导入所需要的库 设置全局参数 图像预处理 读取数据 设置模型 设置训练和验证 测试 完整代码:摘要 EfficientNet是谷歌2019年提出的分类模型,自从提出以后这个模型,各大竞赛平台常常能看到他的身影,成了霸榜的神器。下图是EfficientNet—B0模型的网…

    2021/6/5 10:26:21 人评论 次浏览
  • PyTorch 1.5上线:加入稳定C++前端,高级自动梯度API

    点击上方 "程序员小乐"关注, 星标或置顶一起成长 每天凌晨00点00分, 第一时间与你相约 每日英文 Never expect. Never assume. And never demand. Just let it be. If its meant to be, it will happen. 不要期待,不要假想,不要强求,顺其自然,如果注定,便一…

    2021/6/4 12:23:18 人评论 次浏览
  • windows c++调用yolov5模型 配置及使用

    目录 环境:前期准备:libtorch下载(Release版):opencv下载:cuda10.2环境变量设置 vs配置vs测试 运行yolov5模型一些常见问题无法定位程序输入点***于动态链接库torch_cuda.dll上error :c2872 std 不明确的符号环境: windows 10; vs2019; libtorch1.8.1-cuda10.2 op…

    2021/6/4 7:27:37 人评论 次浏览
  • one of the variables needed for gradient computation has been modified by an inplace operation

    记录一个pytorch多卡训练遇到的bug 报错如下: RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [512]] is at version 30; expected version 29 instead. Hint: the backtrace…

    2021/6/2 18:51:19 人评论 次浏览
  • 简单常用的的环境配置和linux查询命令(用到但想不起来时继续更新补充。。。)

    以下命令仅为了方便自己查阅 1.查看镜像源配置 conda config --show2.添加/删除 镜像源 conda config --add xxx(镜像源地址,如https://pypi.douban.com/simple) #添加 conda config --remove xxx(已有的镜像源地址) #删除3.查看自己的虚拟环境 conda info --env…

    2021/5/31 7:21:13 人评论 次浏览
  • DL Practice:Cifar 10分类

    Step 1:数据加载和处理 一般使用深度学习框架会经过下面几个流程: 模型定义(包括损失函数的选择)——>数据处理和加载——>训练(可能包括训练过程可视化)——>测试 所以自己写代码的时候基本上按照这四大模块四步走就ok了。 本例步骤: A、Load and normal…

    2021/5/24 18:29:27 人评论 次浏览
  • Pytorch学习

    torch.nn.Conv2d 初始化模型:conv = torch.nn.Conv2d(1, 3, kernel_size=3, stride=2, padding=1)第一个参数:表示输入的通道数 第二个参数:表示输出的通道数 第三个参数:表示卷积核大小 第四个参数:表示步幅长度 第五个参数:是否需要padding调用模型:out = conv(x…

    2021/5/23 10:27:12 人评论 次浏览
  • 【编程55--代码share&debug14】Tensorflow,Pytorch 出现 ‘Tensor‘ object is not callable解决办法

    呆呆象呆呆的所有博客目录Tensorflow,Pytorch 出现 ‘Tensor‘ object is not callable解决办法 主要可能的原因: 出现问题的原因1: 可能是在你想调取张量的属性的时候,而写成了函数形式# 原因1对应的代码 import torch tensor = torch.zeros((3,3)) print(tensor.sha…

    2021/5/22 20:29:34 人评论 次浏览
  • PyTorch程序设计

    PyTorch程序设计 1. 程序示例2. 加载数据3. 创建网络(BN、卷积、ReLU)4. 优化器(优化器比较、参数lr、权重衰减)5. 损失函数比较6. 模型保存与加载1. 程序示例 import torch, os from torch.utils.data import Dataset, DataLoader import numpy as npclass MyDataSet…

    2021/5/22 1:55:50 人评论 次浏览
  • 检查cuda和pytorch版本是否匹配

    目的:在已经安装pytorch和cuda的环境下,检查当前版本下pytorch和cuda版本是否兼容 在cmd输入python,运行python 输入 import torch print(torch.__version__) print(torch.cuda.is_available()) 若最终输出为false则表示pytorch和cuda版本不兼容如下图所示:若输出为tr…

    2021/5/10 18:26:39 人评论 次浏览
  • pytorch禁止/允许计算局部梯度/ torch.no_grad()

    版权声明:本文为CSDN博主「Answerlzd」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。 原文链接:https://blog.csdn.net/Answer3664/article/details/99460175

    2021/5/8 10:55:10 人评论 次浏览
  • Pytorch系列:(四)IO操作

    首先注意pytorch中模型保存有两种格式,pth和pkl,其中,pth是pytorch默认格式,pkl还支持pickle库,不过一般如果没有特殊需求的时候,推荐使用默认pth格式保存 pytorch中有两种数据保存方法,一种是存储整个模型,一种只存储参数 方法一:存储整个模型 拼多多优惠券 htt…

    2021/5/7 19:00:36 人评论 次浏览
扫一扫关注最新编程教程