• 深度学习论文: Attention is All You Need及其PyTorch实现

    时间:2024-04-01 15:59:00

    深度学习论文: Attention is All You Need及其PyTorch实现 Attention is All You Need PDF:https://arxiv.org/abs/1706.03762.pdf PyTorch: https://github.com/shanglianl...

  • pytorch之model.eval、model.fuse及model.fuse.eval介绍

    时间:2024-04-01 09:25:25

            在 PyTorch 中,model.eval() 是用于将模型设置为评估模式的方法,而 model.fuse() 是用于量化模型中的融合操作的方法。下面是它们的详细介绍: 1. model.eval()方法介绍         当涉及到 PyTorch 中的模型评估时,model.e...

  • windows10 64位+cpu(无GPU) +Anaconda 安装 pytorch

    时间:2024-03-31 18:49:03

          现在看作者这篇文章的,已经是绕弯路了,最近pytorch推出了支持windows版本的pytorch,可以移步作者的新blog 新链接    如果对tensorflow等框架有简单了解的话,可以pip安装,输入下面两句话。不了解的话可以移步上面的链接   pip3 install htt...

  • 深度学习 环境配置(Tensorflow,Pytorch)多版本CUDA共存

    时间:2024-03-31 18:10:55

    建议在anaconda下创建环境,再进行安装,方便各个环境的管理,以下安装方法在cuda和cudnn都安装好的前提下进行的。Tensorflow安装首先创建环境:conda create -n tf_0.11_gpu python=2.7; tf_0.11_gpu为环境的名称,2.7指定了pytho...

  • 解决Pytorch数据集下载时如MNIST数据下载缓慢

    时间:2024-03-31 13:57:26

    解决Pytorch数据集如MNIST数据下载缓慢数据是深度学习的魂,而在download数据集时,下载了一半就停滞不前……这时看你设置的下载目录,在MNIST里会有两个文件夹:processed和raw打开raw能看到已经下载了的数据集。其实MNIST数据集很小,直接访网站,手动下载数据集:http...

  • 超简单!用pytorch训练自己的数据集

    时间:2024-03-31 13:43:17

    在pytorch官网上的tutoral中,教程给了一个例子,训练一个分类器,下载CIFAR10数据集,用一个简单的CNNC网络训练模型。在我们下载的CIFAR10数据集中,已标签的数据被打包并封装在data文件夹里。我们可以在这个例子的基础上,把自己的数据放到其模型下,并实现一个简易的分类器。首先是...

  • 60分钟入门PyTorch,官方教程手把手教你训练第一个深度学习模型

    时间:2024-03-31 13:03:05

    点击我爱计算机视觉标星,更快获取CVML新技术本文转载自机器之心。近期的一份调查报告显示:PyTorch 已经力压 TensorFlow 成为各大顶会的主流深度学习框架。想发论文,不学 PyTorch 怎么行?那么,入门 PyTorch 深度学习需要多久?PyTorch 的一份官方教程表示:只需要 ...

  • nvidia驱动、cuda和GPU版本pytorch安装

    时间:2024-03-31 12:22:46

    1、nvidia官网:https://www.nvidia.cn/2、找到 驱动程序 -> 所有 NVIDIA驱动程序 找到对应显卡的最新驱动3、根据自己电脑的显卡型号 查找驱动,如 GTX1050 4、找到后,下载安装即可问题:NVIDIA-SMI has failed because yo...

  • 深度之眼Pytorch打卡(十四):Pytorch卷积神经网络部件——卷积层与反卷积层(转置卷积层)

    时间:2024-03-31 11:56:53

    前言  人每天处理的信号中,有超过70%的是视觉信号,所以视觉问题,包括分类,检测,分割、风格转换等等占了深度学习任务中的很大部分。而卷积神经网络是计算机视觉领域当之无愧的霸主。卷积神经网络是稀疏连接,并且权值共享的,参数比全连接要少非常多,所以完完全全可以用图像全像素作为输入,并且它比全连接网络更...

  • 深度学习pytorch——数据增强(持续更新)

    时间:2024-03-31 08:51:21

    背景介绍 大量的数据是防止过拟合的关键,但是我们如何去获取大量的数据,是自己去拍摄、录制吗?显然这种方式有极高的成本。我们可以对同一张图片进行变换得到多张图片,比如原来只有10张图片,通过变换变成了20张图片,但是并不能无限的增长,也是有限制的,这种方法就叫做数据增强。数据增强的成本基本为0,虽然最...

  • 京东开源FastReID:用于真实人物重新识别的Pytorch工具箱

    时间:2024-03-31 08:01:46

     作者信息介绍 FastReID是迄今为止最完整,最高性能的工具箱,支持单个和多个GPU服务器。 最近京东开源了FastReID,作为JD AI Research中广泛使用的对象重新识别(re-id)软件系统。高度模块化和可扩展的设计使研究人员易于实现新的研究思路。友好的可管理系统配置和工程部署功能...

  • WGAN-GP学习笔记(从理论到Pytorch实践)

    时间:2024-03-31 07:56:21

    WGAN相关学习,主要文献 improved of training of WGAN首先我们需要明白一般的GAN数学表达式如下:mathminGmaxDEx∼Pr[log(D(x))]+Ex~∼Pg[log(1−D(x~))]math\mathop{min}\limits_{G}\mathop{ma...

  • Paperreading之三Simple Baselines for Human Pose Estimation(人体姿态coco亚军,pytorch)

    时间:2024-03-31 07:47:33

      本次paper是coco2018关键点检测项目的亚军方案,方法非常的简洁明了,但是效果很惊艳,达到了state of the art,paper的标题也是写了simple baseline。整篇paper包含一个sota的姿态估计和姿态跟踪,本篇博客只涉及到paper里面的姿态估计部分。前言人体...

  • [ 人体姿态估计 ] —— 项目学习: (1) pytorch-pose学习

    时间:2024-03-31 07:47:09

    模型:人体姿态估计沙漏网络(Stacked hourglass network)项目地址:https://github.com/bearpaw/pytorch-pose |--- 几点重要备忘1、LEEDS_annotations.json文件的生成:Use https://github.com/b...

  • Pytorch for training1——read data/image

    时间:2024-03-31 07:05:57

    blog torch.utils.data.Dataset create dataset with class torch.utils.data.Dataset automaticlyimport torchfrom torch.utils.data import Datasetclass MyDa...

  • 【动手学深度学习-pytorch】8.5 循环神经网络的从零开始实现-梯度裁剪

    时间:2024-03-30 17:52:39

    为了防止梯度爆炸或者消失,进行梯度剪裁 @savedef train_epoch_ch8(net, train_iter, loss, updater, device, use_random_iter): """训练网络一个迭代周期(定义见第8章)""" state, timer = ...

  • 使用 PointNet 和 PyTorch3D 进行点云分类-PointNet 的用例

    时间:2024-03-30 09:18:03

    PointNet 是一种高效的架构,支持许多此类应用,包括增强现实、3D 扫描等。它的多功能性和简单性使其能够支持不仅仅是分类的任务,包括(但不限于): 3D 物体检测 点法线的预测 物体部分的分割 语义场景分割PointNet 还具有高度可扩展性,许多已发表的增强或扩展 PointNet 架构的...

  • pytorch学习笔记(十五)————Early Stop,Dropout

    时间:2024-03-29 15:35:28

    pytorch学习笔记(十五)————Early Stop,Dropout,SGD目录Early StopDropoutSGD随机梯度下降目录Early Stop(1)Early Stop的概念Early Stop的概念非常简单,在我们一般训练中,经常由于过拟合导致在训练集上的效果好,而在测试集上的...

  • pytorch 在加载模型时出现的错误

    时间:2024-03-29 11:18:30

    在训练完后,在另一个.py中想导入模型进行使用。导入之后出现这样的错误:pytorch AttributeError: Can’t get attribute ‘Net1’ on <module ‘main’>网上有很多原因, 有一部分是说多线程。 但经过查看文档,这种情况不属于多线程的...

  • 微信开源推理加速工具 TurboTransformers,性能超越 PyTorch/TensorFlow 与主流优化引擎

    时间:2024-03-28 19:17:59

    腾讯宣布开源 Transformer 推理加速工具 TurboTransformers。该工具面向自然语言处理领域中 Transformers 相关模型丰富的线上预测场景,据介绍,其在微信、腾讯云、QQ 看点等产品的线上服务中已经广泛应用,这也是腾讯对外开源的第 100 个项目。在自然语言处理领域中...