torch 相关问题

Torch是LuaJIT的科学计算框架。它被世界各地的机器学习研究人员广泛使用。

将 dlib::array2d<dlib::rgb_pixel> 转换为 torch::Tensor

我使用本教程训练了 torch 模型并将 pth 转换为脚本模块。然后我像这里一样制作了自定义 C++ 项目。修改此示例以输入张量，如下所示： const std::向量我使用本教程训练了 torch 模型并将 pth 转换为脚本模块。然后我制作了自定义 C++ 项目就像这里。修改此示例以输入如下张量： const std::vector<torch::jit::IValue> inputs(1, torch::ones({1, 3, inputSize, inputSize})); auto output = module.forward(inputs).toTensor(); std::cout << "raw output: " << output << '\n' << "soft max: " << torch::softmax(output, 1) << '\n'; NB1 inputSize 大约 128 或 256 像素（导出模块很少）。这个输入有效现在我有了像dlib::array2d<dlib::rgb_pixel>这样的dlib图像。我尝试加载 blob 数据： at::Tensor convert(const dlib::array2d<dlib::rgb_pixel> & img) { auto options = torch::TensorOptions().dtype(torch::kUInt8); std::vector<int64_t> dims = {1, 3, img.nr(), img.nc()}; auto t = torch::from_blob((void*)img.begin(), dims/*, options*/).clone(); return t; } 但是如果我计算出结果张量，有几个数字为 1e+38 或 1e+40；通过模型转发后的结果我在 nan 中有 module.forward(tensorFromConverFunction).toTensor()。 NB2也许一些slice可能有帮助，但我无法找出解决方案。 at::Tensor convert(const dlib::array2d<dlib::rgb_pixel> & img) { const auto height = img.nr(); const auto width = img.nc(); auto tensor = at::zeros({1, 3, height, width}); for (int y = 0; y < height; ++y) { for (int x = 0; x < width; ++x) { tensor[0][0][y][x] = img[y][x].blue; tensor[0][1][y][x] = img[y][x].green; tensor[0][2][y][x] = img[y][x].red; } } return tensor; }

c++ torch dlib

回答 1 投票 0

ModuleNotFoundError：没有名为“torch”的模块，但已安装torch

我正在尝试使用（并已成功安装）布局解析器，它需要 detectorron2 才能实现某些功能。在尝试安装 detectorron2 时，我遇到了以下错误： > pyth...

python installation pip torch

回答 1 投票 0

如果 colab 2024 中 torchtext 导入错误，请尝试此操作

如果您正在 Colab 上工作并尝试导入 torchtext，那么您将收到这样的 torchtext 错误。 OSError：/usr/local/lib/python3.10/dist-packages/torchtext/lib/libtorchtext.so：未定义的符号：

pytorch google-colaboratory torch torchtext

回答 1 投票 0

如何从数组中获取三元组？

我有[1,2,3]，我希望得到[(1,1,1),(1,4,9),(1,8,27)]。实现这一目标的最简单方法是什么？

python python-3.x list numpy torch

回答 3 投票 0

SD3.5-medium 内存消耗巨大

我有一个 g4dn.xlarge AWS GPU 实例，它具有 16GB 内存 + 48GB 交换空间，以及一个具有 16GB vRAM 的 Tesla T4 GPU 实例。根据稳定性博客，运行 SD3.5 Medium 模型应该足够了...

python torch stability diffusers

回答 1 投票 0

torchvision mnist RemoteDisconnected：远端关闭连接无响应

我有一个 pytorch 和 mnist 错误...... 为什么会出现这个错误？ RemoteDisconnected：远端关闭连接无响应进口火炬导入 torchvision.datasets 作为 dsets 小鬼...

python deep-learning torch mnist torchvision

回答 1 投票 0

“torch.distributions.normal.Normal”与“torch.distributions.normal.Normal”之间的任何区别`torch.distributions.Normal`

我只在官方文档中看到torch.distributions.normal.Normal，但从未见过torch.distributions.Normal。然而，我有时看到人们在 Git 的代码中使用 torch.distributions.Normal...

pytorch torch normal-distribution

回答 1 投票 0

如何使用torch.unique过滤重复值，计算一个昂贵的函数，将其映射回来，然后计算梯度？

我正在尝试优化 PyTorch 中的计算，首先识别张量的唯一元素，仅将昂贵的函数（例如 torch.exp）应用于这些唯一元素，然后映射...

pytorch torch automatic-differentiation

回答 1 投票 0

查找每行中连续零的最大长度

我的目标是找到每行中连续零的最大长度。举例来说，如果我有一个像这样的张量输入 = torch.tensor([[0, 1, 0, 0, 0, 1],[0, 0, 1, 0, 1, 0],[1, 0, 0, 0, 0, 0]]) 我...

python numpy pytorch torch

回答 1 投票 0

使用火炬收集图像堆栈的每个图像的不同像素

我有一批图像和每张图像的一批索引（x，y）。每个图像的索引都不同，所以我不能使用简单的索引。获得另一批的最佳或最快方法是什么......

pytorch torch

回答 1 投票 0

Scaled_dot_product_attention 较高的 head num 会消耗更多内存

我发现当头数很大（>=16）时，Scaled_dot_product_attention 会消耗更多的内存。这是我重现该问题的代码。进口火炬长度 = 10000 暗淡 = 64 头数1 = 8 head_num2 ...

pytorch torch transformer-model

回答 1 投票 0

Azure Synapse 中的 Pytorch 导致问题

我在 Azure Synapse 中有一个笔记本正在使用这些库将 pandas 导入为 pd 将 numpy 导入为 np 从 sqlalchemy 导入 create_engine, 文本将 sqlalchemy 导入为 sa 来自 azure.core.credentials

tensorflow pytorch azure-synapse torch

回答 1 投票 0

运行时错误：在当前进程完成引导阶段之前尝试启动新进程

我开发了一个使用 Pytorch 执行神经网络建模的代码。我的代码在 Pycharm 控制台中运行得很好，但当我从脚本运行它时它不起作用。我检查了好几次了...

python torch

回答 2 投票 0

构建CNN模型时出现“顺序对象没有属性权重”错误

从 google.colab 导入驱动器驱动器.mount（'/ gdrive'，force_remount = True）导入操作系统将 numpy 导入为 np 进口火炬将 torch.nn 导入为 nn 从 sklearn.metrics 导入 precision_score 来自 torch.util...

python runtime-error attributeerror torch

回答 1 投票 0

我应该安装哪个版本的torch和torchtext

我使用的是Windows，Python版本为3.11.4，pandas版本为2.2.1 我正在尝试安装 torch 和 torchtext 但总是出现依赖错误。安装的正确版本是什么...

python pytorch torch

回答 1 投票 0

深度强化学习问题：损失减少但智能体不学习

我希望有人能帮助我。我正在为 CartPole-v1 体育馆环境实现基本的香草策略梯度算法，但我不知道我做错了什么。无论我尝试什么，在...

python deep-learning reinforcement-learning torch policy-gradient-descent

回答 1 投票 0

如何为pytorch分配更多内存

我不断出现 Cuda 内存不足错误，我有一个带有 24GB vram 的 3090，torch 只分配 7GB，15GB 始终是空闲的。运行时错误：CUDA 内存不足。尝试分配 92.00 MiB (GPU 0; 24.00...