分类任务使用Pytorch实现Grad-CAM绘制热力图

给大家打个比方,下图是我输出模型结构的一小块,此时如果我设定target_layers=[model.patch_embed_a],那它就相当于被我设置成一个元组,但这样是无法应用在这个代码上的,我们需要继续细化到某一层,比如改成target_layers=[model.patch_embed_a.

vscode终端安装pytorch环境全流程小白版(linux+windows通用版)

在理解本教程的思路后,同理可方便的学会其他情况下的安装思路,并不局限于编辑器是vscode还是pycharm,也不局限于操作系统是linux还是windows。也不局限于大家是在系统默认终端,还是vscode或者pycharm里面的终端进行操作,无所谓,都可以。最好的步骤是按照上述思路一步步明确进行

强化学习highway_env代码解读

作为强化学习的新手,写这个系列的博客主要是为了记录学习过程,同时也与大家分享自己的所见所想。前段时间发布了人生第一篇博客,是关于highway_env的自定义环境。但博客主要是关于如何创建一个自己的环境的py文件,是基于十字路口环境创建的。在经过一段时间的摸索之后,现在基本能理清楚highway_e

【AI】AI入门之Nvidia Jetson平台(四)- Xavier NX软件开发环境安装部署

AI人工智能,深度学习,Jetson嵌入式平台,Ubuntu系统

【Debug记录】深度学习Pytorch+Anaconda环境下常见报错的原因及解决方案 | 亲测已解决

001.“AssertionError: Torch not compiled with CUDA enabled”、 002.“RemoveError: 'requests' is a dependency of conda and cannot be removed from conda's o

No module named 'torch'怎么办

如果在使用 Python 程序时出现 "No module named 'torch'" 错误,说明你的环境中没有安装 PyTorch 库。可以使用以下命令来安装 PyTorch:pipinstall torch如果你正在使用 Anaconda 环境,则可以使用以下命令来安装 PyTorch:con

【代码复现】Windows10复现nerf-pytorch

本文主要介绍了nerf-pytorch在win10下复现的方法。

Ubuntu20.04LTS安装CUDA并支持多版本切换

由于我工作站(Ubuntu 20.04 LTS)的英伟达驱动版本为520.61.05,从上图可以看出,我最高可以安装的CUDA版本为11.8.x。(注:CUDA 12.0.x和CUDA 12.1.x都要求英伟达驱动版本大于等于525.60.13,因此我的520.61.05不符合,所以我最高只能安装C

Linux环境配置mmlab环境

在Linux环境下安装配置mmlab

Pytorch查看GPU信息

GPU,Torch,Cuda

DenseNet网络详解及Pytorch实现

DenseNet是由Gao Huang等研究人员于2017年提出的一种深度神经网络架构。DenseNet的主要思想是在网络的每一层之间建立密集的连接,这种密集连接的结构使得网络在训练过程中可以更好地传播梯度信息,有效地缓解了梯度消失问题。DenseNet在图像分类、物体检测等计算机视觉任务中取得了出

Pytorch Advanced(二) Variational Auto-Encoder

变分自编码 (Variational Autoencoder) 为了让隐层抓住输入数据特性, 而不是简单的输出数据=输入数据,他在隐层中加入随机噪声(单位高斯噪声)(这个过程也叫reparametrize),以确保隐层能较好抽象输入数据特点。有了随机的一张图片之后,我们把他完整的放入模型中,生成了和

pytorch性能分析工具Profiler

PyTorch Profiler 是一个开源工具,可以对大规模深度学习模型进行准确高效的性能分析。分析model的GPU、CPU的使用率各种算子op的时间消耗trace网络在pipeline的CPU和GPU的使用情况Profiler利用可视化模型的性能,帮助发现模型的瓶颈,比如CPU占用达到80%,

ARM架构下银河麒麟V10系统基于CUDA11.8编译PyTorch

公司近期想尝试本地用下模型,只有服务器安装了两张显卡,故而只能在服务器做尝试。CUDA驱动啥的,之前的同事已经安装完毕,并且成功识别出显卡,顾略去。按照GIT README步骤开搞,一切顺利,最后在运行脚本的时候收到如下提示因为服务器是ARM的(CPU是Phytium ST2500)。字面意思,官方

jetson 最全jetpack5.1安装记录,跟着做就行(含bug报错处理)

jetpack5.1使用cuda11和cuda11.4(系统内置了两个),tensorRT8.5.2,内置opencv4.5.4. 虽然有一些bug,但整体是个好版本. 截止23年4月tensorrt8.5依然是最新的稳定版本,想用trt进行加速的小伙伴不要错过.本文会从头开始,从烧录到环境配置来写

【混合精度训练】 torch.cuda.amp.autocast()

【混合精度训练】 torch.cuda.amp.autocast()

PyTorch深度学习实战(15)——迁移学习

迁移学习通过利用相关任务或领域的知识,帮助解决新任务或领域中的学习挑战,可以提高模型的泛化能力、加速模型训练,并在实际应用中取得良好的效果。在图像分类、目标检测、机器翻译等任务中,迁移学习已经展现出巨大的应用价值。在本节中,介绍了迁移学习的基本概念,并使用 PyTorch 构建了迁移学习模型,利用预

人工智能(pytorch)搭建模型10-pytorch搭建脉冲神经网络(SNN)实现及应用

大家好,我是微学AI,今天给大家介绍一下人工智能(pytorch)搭建模型10-pytorch搭建脉冲神经网络(SNN)实现及应用,脉冲神经网络(SNN)是一种基于生物神经系统的神经网络模型,它通过模拟神经元之间的电信号传递来实现信息处理。与传统的人工神经网络(ANN)不同,SNN 中的神经元能够生

【AI】《动手学-深度学习-PyTorch版》笔记(二十):图像增强、微调

微调属于迁移学习的一种,在已经从大量数据集上训练出来的模型上继续训练自己的小数据集,继承已经学习到的边缘、纹理、形状等信息。通过对图像进行随机裁剪,使物体以不同的比例出现在图像的不同位置。

登录可以使用的更多功能哦! 登录
作者榜
...
资讯小助手

资讯同步

...
内容小助手

文章同步

...
Deephub

公众号:deephub-imba

...
奕凯

公众号:奕凯的技术栈