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

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

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

当我将 numpy 数组转换为张量并以不同的方式更改 numpy 时，有些会更改张量，有些则不会。 在此代码中： array = np.arange(1., 8.) tensor = torch....

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

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

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

我遇到了一个问题，即 pytorch（使用 2.0.1+cu117 测试）在发生 CPU OOM 时不会正常失败。具体来说，我失去了所有 ssh 连接和对 VM 或裸机的 Xserver 访问权限。 ...

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

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