问题标签 [pytorch]

For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.

0 投票
3 回答
9567 浏览

python - 从一维张量中提取前 k 个值索引

给定 Torch ( torch.Tensor) 中的一维张量,包含可以比较的值(例如浮点),我们如何提取该张量中前k个值的索引?

除了蛮力方法,我正在寻找一些 API 调用,由 Torch/lua 提供,它可以有效地执行这个任务。

0 投票
8 回答
36916 浏览

python - 如何在 Tensorboard 中显示自定义图像(例如 Matplotlib Plots)?

Tensorboard 自述文件的图像仪表板部分说:

由于图像仪表板支持任意 png,因此您可以使用它将自定义可视化(例如 matplotlib 散点图)嵌入到 TensorBoard 中。

我看到如何将 pyplot 图像写入文件,作为张量读回,然后与 tf.image_summary() 一起使用以将其写入 TensorBoard,但自述文件中的这一声明表明有更直接的方法。在那儿?如果是这样,是否有任何进一步的文档和/或如何有效地做到这一点的例子?

0 投票
1 回答
5157 浏览

python - Python 轮子:不支持 cp27mu

我正在尝试在 Linux 上安装 pytorch ( http://pytorch.org/ ),根据我的机器配置,我应该运行的命令是:

pip install https://s3.amazonaws.com/pytorch/whl/torch-0.1.6.post17-cp27-cp27mu-linux_x86_64.whl

在一台机器上(Linux 发行版 Slackware 14.1)安装失败并出现错误:torch-0.1.6.post17-cp27-cp27mu-linux_x86_64.whl is not a supported wheel on this platform.,而在另一台机器(Ubuntu 15.10)上安装成功。据我了解,问题似乎出cp27mu在车轮名称上。

使用import pip; print(pip.pep425tags.get_supported())Python shell 中的命令,我从 Slackware 机器上得到了这个:

[('cp27', 'cp27m', 'manylinux1_x86_64'), ('cp27', 'cp27m', 'linux_x86_64'), ('cp27', 'none', 'manylinux1_x86_64'), ('cp27', 'none', 'linux_x86_64'), ('py2', 'none', 'manylinux1_x86_64'), ('py2', 'none', 'linux_x86_64'), ('cp27', 'none', 'any'), ('cp2', 'none', 'any'), ('py27', 'none', 'any'), ('py2', 'none', 'any'), ('py26', 'none', 'any'), ('py25', 'none', 'any'), ('py24', 'none', 'any'), ('py23', 'none', 'any'), ('py22', 'none', 'any'), ('py21', 'none', 'any'), ('py20', 'none', 'any')]

这来自Ubuntu机器:

[('cp27', 'cp27mu', 'manylinux1_x86_64'), ('cp27', 'cp27mu', 'linux_x86_64'), ('cp27', 'none', 'manylinux1_x86_64'), ('cp27', 'none', 'linux_x86_64'), ('py2', 'none', 'manylinux1_x86_64'), ('py2', 'none', 'linux_x86_64'), ('cp27', 'none', 'any'), ('cp2', 'none', 'any'), ('py27', 'none', 'any'), ('py2', 'none', 'any'), ('py26', 'none', 'any'), ('py25', 'none', 'any'), ('py24', 'none', 'any'), ('py23', 'none', 'any'), ('py22', 'none', 'any'), ('py21', 'none', 'any'), ('py20', 'none', 'any')]

https://www.python.org/dev/peps/pep-0513/在我看来,支持cp27mcp27mu依赖于编译时传递的选项,--enable-unicode.

现在,也许在这一点上我什至不应该问这个问题,但可以肯定的是,这是否意味着我必须--enable-unicode=ucs4在 Slackware 机器上编译 Python 才能安装那个轮子?

0 投票
2 回答
1537 浏览

macos - 安装 Anaconda 后 PyTorch 不导入

我在安装 Anaconda 后刚刚安装了 PyTorch,当我使用 Anaconda Python 运行 iPython 时,它不会找到 PyTorch。 在此处输入图像描述

在此处输入图像描述

但是我可以验证 PyTorch 是否在我的 anaconda 文件夹的 pkgs 目录中。这是怎么回事?

0 投票
2 回答
16957 浏览

python - 在 PyTorch 中加载 Torch7 训练的模型 (.t7)

我正在使用 Torch7 库来实现神经网络。大多数情况下,我依赖预先训练的模型。在 Lua 中,我使用 torch.load 函数加载保存为 torch .t7 文件的模型。我对切换到 PyTorch( http://pytorch.org )很好奇,我阅读了文档。我找不到有关加载预训练模型的机制的任何信息。我能找到的唯一相关信息是这个页面: http: //pytorch.org/docs/torch.html

但是页面中描述的函数torch.load似乎加载了一个用pickle保存的文件。如果有人有更多关于在 PyTorch 中加载 .t7 模型的信息,请在此处分享。

0 投票
5 回答
45694 浏览

python - PyTorch:如何将 DataLoaders 用于自定义数据集

如何在您自己的数据(不仅仅是)上使用torch.utils.data.Dataset和?torch.utils.data.DataLoadertorchvision.datasets

有没有办法使用DataLoaders他们使用的内置TorchVisionDatasets在任何数据集上使用?

0 投票
0 回答
799 浏览

installation - 我可以在 nvidia jetson tx1 上安装 pytorch 吗?

因为Tensorflow不支持ARM芯片,所以想在Jetson TX1上安装pytorch做一些实验。但是我没有找到任何有关将pytorch构建到Jetson TX的相关资源,这可能吗?

0 投票
9 回答
209231 浏览

python - PyTorch 中的“视图”方法是如何工作的?

view()对以下代码片段中的方法感到困惑。

我的困惑是关于以下行。

函数有什么tensor.view()作用?我在很多地方都看到过它的用法,但我不明白它是如何解释它的参数的。

如果我将负值作为view()函数的参数,会发生什么?例如,如果我打电话,会发生什么tensor_variable.view(1, 1, -1)

谁能view()用一些例子来解释功能的主要原理?

0 投票
11 回答
216617 浏览

python - pytorch 中的模型总结

如何像 Keras 中的model.summary()方法一样在 PyTorch 中打印模型的摘要:

0 投票
1 回答
1030 浏览

amazon-s3 - 安装 pytorch 时出现 HTTP 错误

我使用 ubuntu14.04,没有 GPU。当我在 anaconda 中安装 pytorch 时(我像 pytorch.org 所说的那样输入命令),事情总是出错,就像这样:

我已经尝试了很多次,但每次都在同一步骤中发生错误。期待您的回答,非常感谢您的帮助!!!