pytorch 相关问题

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

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

我正在尝试在我的数据集上实现素数增强。以前我收到错误 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

使用 pytorch 在本地加载 *.pth 检查点

我尝试从本地文件离线加载 VGG19 检查点,而不是常规的 pytorch 方法(在线下载),但遇到了问题。 所以基本上我正在这样做: https://pytorch.org/

回答 1 投票 0

稳定的扩散管始终输出 512*512 图像,无论输入分辨率如何

我正在制作一个修复应用程序,我几乎得到了想要的结果,除了管道对象输出 512*512 图像,无论我传入什么分辨率。我在 CPU 上运行它,它是 .. .

回答 1 投票 0

属性错误:初始化tensorboard SummaryWriter时模块“tensorflow”没有属性“io”

我在尝试将 torch.utils.tensorboard.SummaryWriter 与 PyTorch 结合使用时遇到问题。我正在 torch 中训练 DCGAN,尽管没有在我的代码中直接导入或使用 TensorFlow...

回答 1 投票 0

了解 PyTorch 模型中的批处理

我有以下模型,它构成了我的整个模型管道中的步骤之一: 进口火炬 将 torch.nn 导入为 nn 类 NPB(nn.Module): def __init__(self, d, nhead, num_layers, dropout=0.1):...

回答 1 投票 0

将高维张量映射到低维张量:PyTorch 中的 (B, F, D) -> (B, F-n, D)

我有一个嵌入张量,我想将其减少为更少数量的嵌入。我在批处理环境中工作。张量形状为 B, F, D,其中 B 是批次数,F 是

回答 1 投票 0

PyTorch 1.5.0 CUDA 10.2 通过 pip 安装始终安装 CUDA 9.2

我目前正在安装环境,需要 pytorch 1.5.0 和 CUDA 10.2 。 CUDA 驱动程序已设置,一切正常,但通过 pip 下载 pytorch 已损坏。 之前的官方文档

回答 2 投票 0

如何安装pytorch=1.0以及为什么pytorch安装命令不再起作用

在https://pytorch.org/get-started/previous-versions/中,安装pytorch 1.0的方法是运行“conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch” 。它失败了,因为 pyt...

回答 1 投票 0

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