pytorch

使用LoRA和Hugging Face高效训练大语言模型

在本文中，我们将展示如何使用 大语言模型低秩适配 (Low-Rank Adaptation of Large Language Models，LoRA)...

2023-04-14 标签：PLMAWSLoRa模块 73 0

你需要知道的11个Torchvision计算机视觉数据集

计算机视觉是一个显著增长的领域，有许多实际应用，从自动驾驶汽车到面部识别系统。该领域的主要挑战之一是获得高质量的数据集来训练机器学习模型。

2023-04-10 标签：gpu计算机数据库 66 0

如何应用Anomalib在数据集不平衡的情况下检测缺陷 ？

基于监督式学习的方法利用足够的注释异常样本，通常可用于实现令人满意的异常检测结果。

2023-03-27 标签：机器人加速器数据集 292 0

PyTorch 2.0正式版发布！

作为torch.compile的基础技术，带有Nvidia和AMD GPU的TorchInductor将依赖OpenAI Triton深度学习编译器来生...

2023-03-24 标签：人工智能代码pytorch 245 0

适配PyTorch FX让量化感知训练更简单

近年来，量化感知训练是一个较为热点的问题，可以大大优化量化后训练造成精度损失的问题，使得训练过程更加高效。

2023-03-08 标签：pythonpytorchOneFlow 181 0

100行的Pytorch代码实现三维重建编程分享

渲染是从 3D 模型创建图像的过程。该模型将包含纹理、阴影、阴影、照明和视点等特征，渲染引擎的作用是处理这些特征以创建逼真的图像。

2023-02-28 标签：pytorch 45 0

PyTorch文本分类任务的基本流程

文本分类是NLP领域的较为容易的入门问题，本文记录文本分类任务的基本流程，大部分操作使用了**torch**和**torchtext**两个库。 ...

2023-02-22 标签：数据预处理nlppytorch 157 0

PyTorch指定GPU进行训练

1 直接在终端中设定： CUDA_VISIBLE_DEVICES=1 python main.py 2 python 代码中设定： import oso...

2023-01-13 标签：pythonpytorch 155 0

在DeepStream上使用自己的Pytorch模型

笔者使用的硬件为Jetson Nano 2GB/4GB，参照官方提供的步骤与对应的版本，几乎可以说是无痛安装。对比同样采用干净映像档，使用源码或是Doc...

2022-11-25 标签：C++pytorch 567 0

TensorFlow的衰落与PyTorch的崛起

在采访开发者、硬件专家、云提供商以及熟悉谷歌机器学习工作的人士时，他们的观点也是相同的。TensorFlow 在争夺开发者人心的竞争中落败。其中有些人甚...

2022-11-04 标签：机器学习生态系统pytorch 681 0

新诞生的机器学习框架可以让你更加直观、轻松地拼完整个模型

为了解决这两个挑战，研究人员提出了一个新的基于学习的框架，即 MEPNet（Manual-to-Executable-Plan Network），它从一...

2022-08-08 标签：机器学习开源框架pytorch 358 0

一种全新的数据蒸馏方法来加速NeRF

学术界已有不少研究工作来加速 NeRF。比较流行的一种方式是, 给定训练好的 NeRF, 采用更高效的数据结构进行存储, 如 Sparse Voxel ...

2022-08-08 标签：神经网络数据结构pytorch 480 0

在PyTorch中使用ReLU激活函数的例子

PyTorch已为我们实现了大多数常用的非线性激活函数，我们可以像使用任何其他的层那样使用它们。让我们快速看一个在PyTorch中使用ReLU激活函数的例子：

2022-07-06 标签：函数pythonpytorch 1544 0

PyTorch已为我们实现了大多数常用的非线性激活函数

PyTorch已为我们实现了大多数常用的非线性激活函数，我们可以像使用任何其他的层那样使用它们。让我们快速看一个在PyTorch中使用ReLU激活函数的...

2022-07-06 标签：非线性函数pytorch 458 1