pytorch 相关问题

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

安装mmcv的兼容性问题

我在系统上安装 mmcv 时遇到问题。我正在尝试运行一个需要旧版本 mmcv (1.3.18) 的项目,但是当我安装此版本时,我会遇到 incomp...

回答 1 投票 0

LLM 输出重复本身

我目前正在按照本教程制作一个基本的LLM,它会像莎士比亚一样喷出文本(变压器的完整代码在最后)。我已经到了最后,但是当我训练它并得到一个 ou...

回答 1 投票 0

Torch 稀疏使用环境 CUDA 版本而不是 Torch 的版本

我正在尝试在需要以下内容的集群中运行Python脚本: 火炬==1.7.1 火炬稀疏==0.6.8 问题是,当我安装这两个时,torch 附带的 cuda 版本为 10....

回答 1 投票 0

如何编写栅格图?

我在谷歌colab上使用pytorch。 下面我有一个张量矩阵,这就是例子,实际上矩阵大小约为 50 个神经元和 30,000~50,000 次。 a= 火炬. 张量([[0., 0., 0., 0., 0....

回答 3 投票 0

了解不寻常的 YOLO 标签格式及其对训练的影响

我正在研究固定物体的数据集,其中数据分为训练、测试和验证文件夹以及相应的图像和标签。标签位于文本文件中,内容如下...

回答 1 投票 0

Pytorch 0.4.0:可以通过三种方式在 CUDA 设备上创建张量。他们之间有什么区别吗?

我第三条路失败了。 t3 仍在 CPU 上。不知道为什么。 a = np.random.randn(1, 1, 2, 3) t1 = 火炬.张量(a) t1 = t3.to(火炬.device('cuda')) t2 = 火炬.张量(a) t2 = t2.cuda() t3 = 火炬。

回答 2 投票 0

为什么我收到消息“UserWarning:给定的 NumPy 数组不可写,并且 PyTorch 不支持不可写张量。”?

我正在构建自己的数据集: 类 MyDataset(数据集): def __init__(自身,文件夹): self.folders = 文件夹 def __len__(自我...

回答 1 投票 0

在 128x128 图像上训练 ResNet-18 时,RTX 4060 出现 CUDA 内存不足错误

我在使用 RTX 4060 GPU 在 128x128 图像上训练 ResNet-18 模型时遇到 CUDA 内存不足错误。尽管减少了批量大小,我仍然面临这个问题。下面是详细内容...

回答 1 投票 0

请使用 torch.load 和 map_location=torch.device('cpu')

我正在运行Python程序,但我没有GPU,我该怎么做才能让Python使用CPU而不是GPU? $ python extract_feature.py --data mnist --net checkpoint_4.pth.tar --features 预训练 ...

回答 2 投票 0

在 pytorch 中保存模型时,某些参数未保存

我建立了一个编码器-解码器模型,关注变形变形生成。我能够训练模型并根据测试数据进行预测,但在加载保存的 m 后我预测错误......

回答 1 投票 0

如何优化基于 DETR 的目标检测中的预处理和后处理?

我的问题: 如何减少预处理和后处理的时间? 背景资料 我正在使用 DETR 对视频帧实现对象检测。我的系统处理帧...

回答 1 投票 0

如何使用 PyTorch 保存权重和模型定义?

首先,我使用pytorch定义了一个模型并保存了它。 SimpleNN 类(nn.Module): def __init__(自身): 超级(SimpleNN, self).__init__() self.flatten = nn.Flatten() 自我.fc ...

回答 1 投票 0

PyTorch:保存问题,如何保存权重和模型定义

首先,我使用pytorch定义了一个模型并保存了它。 SimpleNN 类(nn.Module): def __init__(自身): 超级(SimpleNN, self).__init__() self.flatten = nn.Flatten() 自我.fc ...

回答 1 投票 0

实现 Pass2Edit 来建模字符串编辑行为

我正在尝试实现Pass2Edit(本文:阅读3.1、3.2)。它接收原始密码和当前密码字符串,并尝试对编辑行为进行建模。下面是它的样子...

回答 1 投票 0

torch.nn.function 与 torch.nn - Pytorch

在 Pytorch 中添加损失时,我在 torch.nn.Functional 和 torch.nn 中具有相同的功能。有什么区别? torch.nn.CrossEntropyLoss() 和 torch.nn.function.cross_entropy

回答 1 投票 0

pytorch中的mask_fill在这里做什么

我有一个名为“k1”的张量,其形状为 3,1,1,9,还有 p1 张量,其形状为 3,7,9,9,我想知道下面的行是做什么的? p1 = p1 .masked_fill(k1== 0, float("-1e30...

回答 1 投票 0

为什么我们要做批量矩阵-矩阵乘积?

我正在关注 Pytorch seq2seq 教程,并使用 ittorch.bmm 方法,如下所示: attn_applied = torch.bmm(attn_weights.unsqueeze(0), 编码器输出.unsqueeze(0)) 我明白...

回答 3 投票 0

无法使用pytorch中的自定义损失函数更新潜在向量

我正在尝试实现这个功能,但没有成功。我正在使用一个 VAE 模型,同时还有编码器和解码器。我正在冻结 VAE 解码器的权重,并且

回答 1 投票 0

用不同的损失训练模型的不同阶段

我正在尝试以端到端的方式训练两阶段模型。但是,我想更新具有不同损失的模型的不同阶段。例如,假设端到端模型由...组成

回答 1 投票 0

如何在具有相同 CUDA 的虚拟环境中设置 TF 和 Torch

我想在一个具有相同 CUDA 的虚拟环境中设置 TensorFlow 和 pytorch。但是,我找不到可以同时支持tensorflow和pytorch的CUDA版本:对于tensorflow 2.10,我选择了...

回答 1 投票 0

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