pytorch 相关问题

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

Navi10 上配备 Pytorch 的 AMD ROCm(RX 5700 / RX 5700 XT)

我是拥有 AMD GPU(RX 5700、Navi10)的悲惨生物之一。我想使用最新的 PyTorch 库在本地计算机上进行一些深度学习并停止使用云实例。 我...

回答 2 投票 0

导入torch.utils.tensorboard会导致tensorflow警告

如此处所述,张量板是张量流的一部分,但不依赖于它。可以在 pytorch 中使用它,例如 从 torch.utils.tensorboard 导入 SummaryWriter 然而,令人烦恼的是,我...

回答 1 投票 0

torch.rand(1, 3, 64, 64) 是什么意思?

我是 PyTorch 的初学者。在一个教程中,我看到:torch.rand(1, 3, 64, 64),我知道它创建了一个具有遵循标准正态分布的随机数的张量。 输出看起来像:

回答 3 投票 0

pytorch - 灰度图像分类模型上的输入和模型大小错误错误

我有 (650, 650) 带标签的灰度图像,并尝试构建 pytorch CNN 模型。 我想快完成了,但是下面有错误。 运行时错误:给定组=1,大小权重[32,1,3,3],预期...

回答 1 投票 0

如何在 Microsoft Azure 上部署 Ultralytics YOLOv8?

有关于如何在 GCP 和 AWS 上部署 Ultralytics YOLOv8 的非常简单的快速入门指南: 谷歌云深度学习虚拟机:https://github.com/ultralytics/yolov5/wiki/GCP-Quickstart 亚马逊深乐...

回答 2 投票 0

我正在尝试使用 YouTube 指南在 Python 中构建 AI 图像分类器。当我运行我的程序(未完成)时,它不会打开图像

我正在尝试使用 YouTube 指南为学校项目构建人工智能图像分类器。这是链接:https://www.youtube.com/watch?v=oEKg_jiV1Ng&t=727s 在这个阶段,我还没有完成,但是...

回答 1 投票 0

关于DDP和单卡训练的batchsize和学习率设置问题

关于DDP和单卡训练的batchsize和学习率设置问题 单卡网络训练,batchsize=8,学习率=10e-4 现在改为DDP单机多...

回答 1 投票 0

我正在尝试使用 YouTube 指南构建人工智能图像分类器。当我运行我的程序(未完成)时,它不会打开图像。 (Python)

我正在尝试使用 YouTube 指南为学校项目构建人工智能图像分类器。这是链接:https://www.youtube.com/watch?v=oEKg_jiV1Ng&t=727s 在这个阶段,我还没有完成,但是...

回答 1 投票 0

用于预训练视觉变压器的自定义补丁嵌入层

我正在寻找一种为普通 ViT 使用自定义补丁嵌入层的方法。我想要预训练模型中的其余 ViT。有没有办法使用 Pytorch 来做到这一点? 我可以加载预先训练好的...

回答 1 投票 0

Neural A*环境安装

我有 ubuntu 23.04,我正在尝试安装 e 并按照说明从 https://github.com/omron-sinicx/neural-astar?tab=readme-ov-file 执行 Neural A*。我有 python 3.12.1 及之后...

回答 1 投票 0

简单的ResNet模型无法判断两个单调图像是否颜色相同

我正在尝试训练图像比较模型的问题。我将其简化为以下问题。 我为模型提供了成对的图像 (3x128x128)。图像要么全黑...

回答 1 投票 0

Pytorch求和问题(可能是浮点数)

我有一个简单的Python测试来获取我放入火炬张量的列表中的几个项目的总和: lst = [0.0014, -0.0306, 0.0005, 0.0011, 0.0012, 0.0022, 0.0017, 0.0011, 0....

回答 1 投票 0

我已经使用了 detach().clone().cpu().numpy() 但仍然引发 TypeError: can't conversion cuda:0 device type tensor to numpy

该函数第 7 行出现错误 def Visualize_embedding(h, color, epoch=None, loss=None): plt.figure(figsize=(7,7)) plt.xticks([]) plt.yticks([]) h = h.detach().clone().cpu().nump...

回答 1 投票 0

对火炬数据集的迭代未加载多个目标

我正在尝试从文件加载数据集并在其上训练人工智能模型。 由于某种原因,当我使用 iamges 时,我的主数据加载器中的目标会加载如下目标: { 'image_id':[所有图像 ID...

回答 1 投票 0

OCR 结果不一致:训练和测试期间的预测不同

我的自定义 OCR(光学字符识别)模型遇到问题,它在训练和测试阶段产生不同的预测。 在训练过程中,模型似乎...

回答 1 投票 0

将“llama-2”8 位量化版本加载到 GPU 上

基于此存储库 GitHub 链接,我正在尝试构建一个回答用户查询的系统。 我能够在 CPU 上运行模型,响应时间约为 60 秒,现在我想提高响应时间,s...

回答 2 投票 0

将 Pytorch 张量移动到 GPU 时出现 PyTorch 分段故障(核心转储)

我有一台配备 RTX 6000 ADA GPU 的机器。 我们曾经使用 CUDA 版本 11.x,我使用了以下图像: nvcr.io/nvidia/pytorch:21.04-py3 (我使用 PyTorch 1.x)。 然而,我们的

回答 1 投票 0

简单的 Torch 模型测试:ModuleNotFoundError:没有名为“ultralytics.yolo”的模块

我有一个模型 best.pt,我想运行它。它以图像作为输入,并输出一个字符串。 我安装了 ultralytics、torch 和 torchvision。 我的代码很简单: 进口火炬 从 PIL 进口...

回答 1 投票 0

如何使用 pytorch lighting 在 AWS 云 GPU 上进行训练?

我目前正在做一个项目,并使用 pytorch lighting 编写 ML 模型。我正在训练的数据集相当大,因此无法在我的本地 GPU 上进行训练。为此,我...

回答 1 投票 0

使用 torch.where() 处理一批数组

我正在使用 pytorch,我想在不使用循环的情况下将简单的 torch.where(array > 0) 应用于一批数组,我如何使用 torch 函数来执行此代码? def batch_node_indices(states_ba...

回答 1 投票 0

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