我有一个 3d 张量，我想使用 .是否要屏蔽某行取决于布尔列表，其 True 值对应于要屏蔽的行。当列表中有一个 True 时，下面的代码按预期工作，但当有两个 True 时会给出一个 IndexEr...

这是从磁盘加载的数据 wt_emb = torch.load("train/train_wt.pt") mut_emb = torch.load("train/train_mut.pt") df =...

对不起，我的英语不好。 我的操作系统是 WIN11，我使用 Anaconda 创建了一个虚拟环境，python 版本是 3.8.18，并使用 conda install pytorch==1.10....

我正在尝试在 pytorch 中的多个 GPU 之间使用张量并行化，特别是 2 个 Nvidia A100，以使用基于 slurm 的 HPC 将一个 GPU 太大的模型传播到多个 GPU 中。我的模...

我有复杂的数据，我想用它来训练火炬中的网络。 我已经设法创建了一个网络和所有张量。 但是，一旦我开始训练，就会收到以下错误：grad can be implicitly created only f...

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

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

我一直在尝试为个人项目重新创建 Dino V1 训练设置。为此，我从这个存储库中获取了大部分代码：https://github.com/facebookresearch/dino[dinov1 lin...

我有以下问题：我在 PyTorch 中开发了一个深度学习管道（但 Tensorflow 也存在这个问题），它适用于我的 GPU 和 CUDA 版本的计算机。没关系。但是现在我希望其他人使用他们的 GP...

我正在尝试使用火炬分布式包进行分布式 cpu 训练。在我的代码中，我通过调用 ddp_setup（） 初始化和创建组，如下所示： def get_dist_info(): GLOBAL_WORLD...