pytorch 相关问题

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

Tensorflow 相当于 torch.scatter_add

如何用tf 1.15实现同样的操作? 进口火炬 B、T、N、K = 2,3,4,2 # a 是一个计数器表,其中 T 是组数 a = torch.zeros(T, N, dtype=torch.long) # x 是一只蝙蝠...

回答 1 投票 0

在 pytorch 中初始化模型时出现问题

我无法在 pytorch 中初始化我的模型并获取: 类型错误回溯(最近一次调用最后一次) 在 () 288 数据集 =

回答 2 投票 0

在pytorch中加载测试数据

如何在 pytorch 中加载我自己的测试数据(image.jpg)来测试我的 CNN?

回答 2 投票 0

如何在 BERT 中获得“正确”的嵌入层?

我想利用 BERT 来评估两段文本之间的相似度: 从 Transformer 导入 AutoTokenizer、AutoModel 进口火炬 导入 torch.nn.function 作为 F 将 numpy 导入为 np 托尼尼...

回答 1 投票 0

数据加载器冻结

自定义数据集上的我的 Pytorch (1.11.0) 数据加载器偶尔会冻结。 我无法重现冻结,它似乎是随机的:它通常“运行”没有问题,但有时会卡住。 W...

回答 1 投票 0

在 pytorch 中重塑张量会产生奇怪的行为

我正在通过 https://github.com/parrt/fundamentals-of-deep-learning/blob/main/notebooks/3.train-test-diabetes.ipynb 作为练习,但忘记重塑 y 张量在这些行中 y_train = t...

回答 1 投票 0

获取用于文本分类的 Captum 文本解释时出错

我有以下代码,用于识别用于正确预测测试数据集中文本的最有影响力的单词 将 pandas 导入为 pd 进口火炬 从 torch.utils.data 导入

回答 1 投票 0

未设置CUDA_HOME环境变量

我有一个使用 pytorch 深度学习和 GPU 的工作环境,当我尝试使用 mmcv.ops.point_sample 时遇到了问题,它返回: ModuleNotFoundError:没有名为“mmcv”的模块...

回答 2 投票 0

为什么在 pytorch 中批量计算比 for 循环花费更多时间?

我的代码有问题吗? 当我一起运行以下代码时,所花费的时间有很大差异: 批处理时间:1.3649392127990723 时间:0.7864551544189453 当我运行它们时

回答 1 投票 0

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

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

回答 1 投票 0

“NotImplementedError:无法使用来自‘QuantizedCPU’后端的参数运行‘aten::add.out’”同时使用 pytorch 在 resnet18 上实现 QAT

我正在尝试实现量化感知训练(QAT)resnet18模型。在推断时我收到此错误 NotImplementedError:无法使用“QuantizedCPU”中的参数运行“aten::add.out”

回答 1 投票 0

运行时错误:r.nvmlDeviceGetNvLinkRemoteDeviceType_内部断言失败于

我正在编写一个Python代码,用于训练分类器对样本进行分类(每个样本10个句子)。我正在使用具有附加层的 Sentence_Transformer 并在 Linux 上运行模型训练...

回答 1 投票 0

尝试计算离线 RL 的策略梯度时出现 Pytorch .backward() 错误

每当尝试使用内存缓冲区来训练我的策略时,我总是会遇到向后传递错误或就地自动分级错误。 这是导致问题的函数: def 强化(pi,

回答 1 投票 0

sklearn 的 r2_score() 和 PyTorch 的 MSELoss() 之间的差异?

我不确定我是否遗漏了一些非常基本的东西,但我开始注意到 sklearn 的 r2_score() 函数返回的 R2 分数和从 P 计算的 R2 分数之间存在一些细微的差异...

回答 1 投票 0

将 PyTorch 转换为 CoreML

我有这个模型:https://github.com/williamyang1991/DualStyleGAN 并尝试将其转换为 CoreML。到目前为止,我创建了原始 Colab 笔记本的副本,并在末尾附加了两个块: !pip 安装

回答 2 投票 0

我正在尝试创建多尺度 CNN,但面临此错误:RuntimeError: mat1 和 mat2 形状无法相乘(32x4095 和 4096x4096)

这是我的模型: 进口火炬 将 torch.nn 导入为 nn 导入 torch.nn.function 作为 F # 定义浅层 CNN 类ShallowCNN(nn.Module): def __init__(self, in_channels, out_dim): 超级(

回答 1 投票 0

在 PyTorch 中使用 DataLoaders 进行 k 倍交叉验证

我已将训练数据集分成 80% 的训练数据和 20% 的验证数据,并创建了 DataLoaders,如下所示。但是我不想限制我的模型的训练。所以我想到了分割我的d...

回答 4 投票 0

哪个设备上有一个包含在 cuda 上加载的 pytorch 张量的 python 字典?

我有一个 pytorch 人脸检测模型,它返回边界框和一些面部标志作为字典。边界框和地标是移动到 GPU 的 pytorch 张量。 W...

回答 1 投票 0

将张量拆分为张量列表的最快方法

假设我有一个形状为 [A,B,C,...] 的张量,并且我希望创建沿第 0 维的张量列表。这意味着输出应该是一个长度为 A 的一维列表,其中每个元素都是一个十......

回答 1 投票 0

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

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

回答 1 投票 0

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