我正在尝试计算具有马氏距离而不是 L1 或 L2 范数的 KNN 邻居 我遇到了将 Tensor 与 Lazy Tensor 相乘的错误。 任何想法如何使用已经解决这个问题的惰性张量或 gpu 库...

我是 PyTorch 的菜鸟，我正在尝试训练我的模型（估计两个图像之间的仿射变换）。 该代码在 google colab 上运行。当我尝试在 cpu（没有设备 cuda）上运行它时，它可以工作，但是当...

我正在尝试创建一个包含 10000 个样本（8000 个用于训练，2000 个用于验证）的 3D 测试数据集来测试我的 3D CNN 模型。看起来一切正常，直到我尝试查看我使用的第一批数据。更具体地说...

我想用训练我自己的自定义模型来制作一个 PyTorch 程序。使用 RIFE 预训练模型（链接：文本）有一个很好的程序。但我想将模型替换为我自己的模型，或者微调预训练的数据集。我不知道如何组织自定义模...

我在我的架构中添加了三层转换器，任务是压缩。训练验证和损失在训练期间工作得很好，而在推理过程中，情况很糟糕，准确性显着下降。我不确定这是否与过拟合有关，但情节至少没有显示（如果我没错的话）。变压器怎么...

我正在尝试使用 Torch 和 CUDA。到目前为止，使用我能够做到以下几点：torch::from_blob() #include <cuda_runtime.h> #include <torch...

为简单起见，我想使用此代码将火炬模型的所有参数设置为常量72114982 model = Net() params = model.state_dict() for k, v in params....

我有一个 Pytorch 程序，可以动态生成 CIFAR-10 的自定义版本（它不会保存自定义的 CIFAR-10，但该程序会下载原始数据集，对其进行自定义，并将其用于训练）。我想将自定义的 CIFA...

我正在使用 PyTorch 从头开始实现 ResNet 网络。这个网络对我的要求是独一无二的，因为我需要对具有 14 个不同通道和尺寸为 8x8 像素的卫星影像执行图像分类。 然而，我的问题是，在训...

我已经为在某个数据集上训练的自定义转换器编写了代码。我的问题在于模型没有利用我的 gpu，即使我已经根据我所知初始化了代码中的所有内容。有人可以帮我弄清楚为什么使用cpu进行训练吗？我使用 kaggl...