草庐IT

pytorch_model

全部标签

Pytorch报错解决——(亲测有效)RuntimeError: Distributed package doesn‘t have NCCL built in

在我跑SLADD模型的时候,出现了如下报错:上网搜寻一番后,发现了解决方法第一步:在本文件的开头机上这样两行代码:importosos.environ["PL_TORCH_DISTRIBUTED_BACKEND"]="gloo"第二步:在本文件代码中找到这样一部分代码:把dist.init_process_group(backend='nccl',init_method='env://',world_size=args.world_size,rank=rank)这一行中的nccl换成gloo若要解释原因就是windows系统不支持NCCL,只能换用GLOO亲测这个报错就没了,然而接下来还有好多

吴恩达AIGC《How Diffusion Models Work》笔记

1.IntroductionMidjourney,StableDiffusion,DALL-E等产品能够仅通过Prompt就能够生成图像。本课程将介绍这些应用背后算法的原理。课程地址:https://learn.deeplearning.ai/diffusion-models/2.Intuition本小节将介绍扩散模型的基础知识,探讨扩散模型的目标,如何利用各种游戏角色图片训练数据来增强模型的能力。假设下面是你的数据集,你想要更多的在这些数据集中没有的角色图片,如何做到?可以使用扩散模型生成这样的角色图片。扩散模型应该是这样的一个神经网络:它能够学习到游戏角色的一般概念,例如游戏角色是什么,游

再探pytorch的Dataset和DataLoader

本文参加新星计划人工智能(Pytorch)赛道:https://bbs.csdn.net/topics/613989052本文从分类、检测、分割三大任务的角度来剖析pytorch得dataset和dataloader源码,可以让初学者深刻理解每个参数的由来和使用,并轻松自定义dataset。思考:在探究Dataset和DataLoader之前,需要明白一个事情,就是当我们不管做是分类、检测还是分割任务时,我们的数据集一定由很多张图片组成的,形状大小各异;那麽我们在使用pytorch时,图片是怎么以一个batch的形式进行打包的呢,形状不同怎么处理,数据格式有什么要求,Dataset类中的初始化

再探pytorch的Dataset和DataLoader

本文参加新星计划人工智能(Pytorch)赛道:https://bbs.csdn.net/topics/613989052本文从分类、检测、分割三大任务的角度来剖析pytorch得dataset和dataloader源码,可以让初学者深刻理解每个参数的由来和使用,并轻松自定义dataset。思考:在探究Dataset和DataLoader之前,需要明白一个事情,就是当我们不管做是分类、检测还是分割任务时,我们的数据集一定由很多张图片组成的,形状大小各异;那麽我们在使用pytorch时,图片是怎么以一个batch的形式进行打包的呢,形状不同怎么处理,数据格式有什么要求,Dataset类中的初始化

Pytorch实现GCN(基于PyTorch实现)

文章目录前言一、导入相关库二、加载Cora数据集三、定义GCN网络3.1定义GCN层3.1.1邻接矩阵A3.1.2度矩阵D3.1.3获取矩阵D−1/2D^{-1/2}D−1/23.1.4GCNConv层3.2定义GCN网络四、定义模型五、模型训练六、模型验证七、结果完整代码前言大家好,我是阿光。本专栏整理了《图神经网络代码实战》,内包含了不同图神经网络的相关代码实现(PyG以及自实现),理论与实践相结合,如GCN、GAT、GraphSAGE等经典图网络,每一个代码实例都附带有完整的代码。正在更新中~✨🚨我的项目环境:平台:Windows10语言环境:python3.7编译器:PyCharmPy

Pytorch实现GCN(基于PyTorch实现)

文章目录前言一、导入相关库二、加载Cora数据集三、定义GCN网络3.1定义GCN层3.1.1邻接矩阵A3.1.2度矩阵D3.1.3获取矩阵D−1/2D^{-1/2}D−1/23.1.4GCNConv层3.2定义GCN网络四、定义模型五、模型训练六、模型验证七、结果完整代码前言大家好,我是阿光。本专栏整理了《图神经网络代码实战》,内包含了不同图神经网络的相关代码实现(PyG以及自实现),理论与实践相结合,如GCN、GAT、GraphSAGE等经典图网络,每一个代码实例都附带有完整的代码。正在更新中~✨🚨我的项目环境:平台:Windows10语言环境:python3.7编译器:PyCharmPy

UserWarning: __floordiv__is deprecated, and its behavior will change in a future version of pytorch.

报错提示:UserWarning:__floordiv__isdeprecated,anditsbehaviorwillchangeinafutureversionofpytorch.Itcurrentlyroundstoward0(likethe‘trunc’functionNOT‘floor’).Thisresultsinincorrectroundingfornegativevalues.Tokeepthecurrentbehavior,usetorch.div(a,b,rounding_mode=‘trunc’),orforactualfloordivision,usetorch.di

Pytorch机器学习(八)—— YOLOV5中NMS非极大值抑制与DIOU-NMS等改进

Pytorch机器学习(八)——YOLOV5中NMS非极大值抑制与DIOU-NMS等改进目录Pytorch机器学习(八)——YOLOV5中NMS非极大值抑制与DIOU-NMS等改进前言一、NMS非极大值抑制算法二、Hard-NMS非极大值代码三、DIOU-NMS 四、soft-NMS前言在目标检测的预测阶段时,会输出许多候选的anchorbox,其中有很多是明显重叠的预测边界框都围绕着同一个目标,这时候我就可以使用NMS来合并同一目标的类似边界框,或者说是保留这些边界框中最好的一个。如果对IOU等知识不了解的可以看我上篇博客Pytorch机器学习(五)——目标检测中的损失函数(l2,IOU,G

Pytorch机器学习(八)—— YOLOV5中NMS非极大值抑制与DIOU-NMS等改进

Pytorch机器学习(八)——YOLOV5中NMS非极大值抑制与DIOU-NMS等改进目录Pytorch机器学习(八)——YOLOV5中NMS非极大值抑制与DIOU-NMS等改进前言一、NMS非极大值抑制算法二、Hard-NMS非极大值代码三、DIOU-NMS 四、soft-NMS前言在目标检测的预测阶段时,会输出许多候选的anchorbox,其中有很多是明显重叠的预测边界框都围绕着同一个目标,这时候我就可以使用NMS来合并同一目标的类似边界框,或者说是保留这些边界框中最好的一个。如果对IOU等知识不了解的可以看我上篇博客Pytorch机器学习(五)——目标检测中的损失函数(l2,IOU,G

LLM - Hugging Face 工程 BERT base model (uncased) 配置

欢迎关注我的CSDN:https://spike.blog.csdn.net/本文地址:https://blog.csdn.net/caroline_wendy/article/details/131400428BERT是一个在大量英文数据上以自监督的方式预训练的变换器模型。这意味着它只是在原始文本上进行预训练,没有人以任何方式对它们进行标注(这就是为什么它可以使用大量公开可用的数据),而是用一个自动的过程来从这些文本中生成输入和标签。更准确地说,它是用两个目标进行预训练的:掩码语言建模(MaskedLanguageModeling,MLM):给定一个句子,模型随机地掩盖输入中的15%的词,然