pytorch 相关问题

PyTorch是一个深度学习框架,它实现了一个动态计算图,它允许您改变神经网络在运行中的行为方式,并能够执行向后自动区分。

训练自定义稳定扩散模型时如何避免内存耗尽错误?

我在训练自定义 LCM-LoRA 模型时遇到此错误。 我使用 347 个自定义图像来训练它。 我尝试使用 2 个 RTX-4090 GPU,但遇到了同样的错误。 我是否必须增加 GPU 数量或使用...

回答 1 投票 0

IndexError:在 PyTorch 训练步骤中,“索引 1 超出了尺寸为 1 的维度 0 的范围”

我目前正在使用 PyTorch 执行对象检测任务,并在训练步骤中遇到“IndexError”。错误信息如下: IndexError:索引 1 超出...

回答 1 投票 0

(Pytorch) mat1 和 mat2 形状不能相乘(212992x13 和 1280x3)

我正在尝试使用自定义数据集在 Pytorch 预训练模型上进行迁移学习。 目前,我收到错误如下 mat1 和 mat2 形状不能相乘(212992x13 和 1280x3) 期间...

回答 1 投票 0

视觉变压器:运行时错误:mat1和mat2形状不能相乘(32x1000和768x32)[关闭]

我正在尝试对视觉变压器模型进行回归,但无法用回归层替换最后一层分类 类 RegressionViT(nn.Module): def __init__(自我,

回答 1 投票 0

沿轴以可变长度索引索引到火炬张量

我正在尝试根据语言模型计算标记化单词列表的单词概率,并且我需要一些花哨的索引。 我的输入,用下面的玩具示例进行说明: 令牌列表:

回答 1 投票 0

LeNet5 和自组织映射 - RuntimeError:尝试再次向后浏览图表 - PyTorch

我进行了 LeNet-5 CNN 训练,并使用在 MNIST 数据上训练的自组织图进行训练。训练代码(为了简洁起见)是: # SOM(扁平)权重- # m = 40, n = 40, n = 84 (LeNet 的输出形状/dim)

回答 1 投票 0

采用 Huggingface 模型的编码器-解码器

我想使用以下结构创建一个编码器-解码器模型: Bert-base-uncased 用于对输入进行编码(https://huggingface.co/google-bert/bert-base-uncased) 用于连接的线性层...

回答 1 投票 0

“图像”对象没有属性“形状”

我正在尝试在我的数据集上实现素数增强。以前我收到错误 raise ValueError('pic should be 2/3 Dimension. Got {}Dimensions.'.format(pic.ndim)) ValueError:图片显示...

回答 1 投票 0

Starcoder 微调 - 如何选择 GPU 以及如何估计微调所需的时间

我想在我的数据集和 GCP VM 实例上微调 Starcoder (https://huggingface.co/bigcode/starcoder)。 文档中说,为了训练模型,他们使用了 512 Tesla A10...

回答 1 投票 0

视觉 Transformer 模型的回归

我正在尝试对视觉变压器模型进行回归,但无法用回归层替换最后一层分类 当我尝试初始化模型时出现此错误 CL...

回答 1 投票 0

我的 PyTorch 回归机器学习程序没有学习

我正在努力使用 PyTorch 创建一个机器学习项目,我意识到我的模型似乎无法学习 - 它总是输出一条几乎没有变化的平坦线,并且损失几乎没有减少。在...

回答 1 投票 0

将 PyTorch CUDA 张量转换为 NumPy 数组

如何将 torch.Tensor(在 GPU 上)转换为 numpy.ndarray(在 CPU 上)?

回答 4 投票 0

torch.stack([t1,t1,t1],dim=1) 和 torch.hstack([t1,t1,t1]) 有什么区别?

从技术上讲,torch.stack([t1,t1,t1],dim=1) 和 torch.hstack([t1,t1,t1]) 方法执行相同的操作,即它们都水平堆叠向量。 但是当我在同一台上表演时

回答 2 投票 0

如何从 Pytorch-Geometric GAT 模型中提取图形节点嵌入?

Dataset Strucute:时间有向图;节点有特性;边缘没有特征;节点被标记。使用椭圆数据集 任务:对节点进行分类/预测节点标签。 数据结构:...

回答 1 投票 0

哪个库可以替代机器学习编程中的 causal_conv1d?

最近,我一直在使用 causal_conv1d 库进行机器学习编程,而 causal_conv1d 是 mamba_ssm 库的一部分。但是,我只能在 NVIDIA GPU 上运行这些库......

回答 1 投票 0

如何在 Visual Studio 2022 中设置 CMAKE_PREFIX_PATH?

我想在 Visual studio 2022 中使用 Pytorch c++ ,这就是为什么我需要在 Visual studio 中设置 CMAKE_PREFIX_PATH ,但我不能。我尝试按照教程 https://pytorch.org/cppdocs 进行操作/

回答 1 投票 0

nn.Conv2d pytorch中out_channel的含义

我正在研究卷积神经网络并创建一个模块。 使用 nn.conv2d 时,我无法弄清楚如何指定我的 out_channels。 我理解是过滤器的数量...

回答 1 投票 0

Tensorflow + Pytorch 安装 Cudatoolkit 11.2

我有一台带有 nvidia 3080 卡的 Windows 10 机器。 11.2 使用CudaToolkit 除了 TensorFlow 之外,我还想安装 Pytorch,到目前为止它运行得 100% 正常。 如果我理解描述正确...

回答 1 投票 0

Pytorch 与 Tensorflow CUDA 版本

我目前安装了 Pytorch(版本 1.13.1+cu116)。它检测 GPU 并与 CUDA 版本 11.6 配合良好。以下是我运行 nvidia-smi 的输出: 我现在正在尝试设置张量流...

回答 2 投票 0

如何正确配置带有线性层的 LSTM 以进行唤醒词检测

我正在为我的人工智能助手开发唤醒词模型。我的模型架构包括一个用于处理音频数据的 LSTM 层,后面是一个线性层。然而,我遇到了意想不到的事情......

回答 1 投票 0

© www.soinside.com 2019 - 2024. All rights reserved.