Home torchvision

torchvision

IT技术

CPU only pytorch 运行时崩溃，报错 AssertionError: Torch not compiled with CUDA enabled

· 2025年5月31日 · 0 Comment

我尝试运行这个仓库中的代码，并且需要使用 Pytorc…

IT技术

AttributeError: 在DataLoader工作进程0中捕获到AttributeError

· 2025年5月31日 · 0 Comment

我已经进行了一些谷歌搜索，这个错误似乎经常出现，但我不…

IT技术

TypeError: __array__() 接受 1 个位置参数，但提供了 2 个

· 2025年5月31日 · 0 Comment

我在做 PyTorch 教程（https://pyto…

IT技术

pytorch torchvision.datasets.ImageFolder FileNotFoundError: 在类.ipynb_checkpoints中找不到有效文件

· 2025年5月31日 · 0 Comment

我在Colab中尝试使用pytorch的torch.d…

IT技术

如何使用torch.hub.load加载本地模型？

· 2025年5月31日 · 0 Comment

我需要避免从网络上下载模型（因为安装机器上的限制）。 …

IT技术

在可变尺寸图像中检测固定尺寸对象

· 2025年5月30日 · 0 Comment

神经网络可以被训练识别一个对象，然后在图像中检测该对象…

IT技术

如何使用PyTorch中的Mask R-CNN预测生成图像的精确掩码？

· 2025年5月29日 · 0 Comment

我已经训练了一个用于苹果实例分割的Mask RCNN网…

IT技术

如何使用预训练模型的权重来初始化下一轮迭代的权重？

· 2025年5月28日 · 0 Comment

我有一个模型架构。我已经使用 torch.save()…

IT技术

如何使用PyTorch在自定义图像数据集中创建训练-验证集划分？

· 2025年5月27日 · 0 Comment

我想从我的原始训练集中创建一个训练+验证集。目录分为训…

IT技术

VGG16在全局平均池化(GAP)之前使用最大池化？

· 2025年5月26日 · 0 Comment

我目前正在使用带有全局平均池化(GAP)的VGG16模…

IT技术

PyTorch – BCELoss: ValueError: 目标和输入必须具有相同数量的元素

· 2025年5月24日 · 0 Comment

当我使用 BCELoss 作为神经网络的损失函数时，遇…

IT技术

torchvision.datasets.cifar.CIFAR10是否是一个列表？

· 2025年4月13日 · 0 Comment

当我试图了解torchvision.datasets….

IT技术

pytorch dataloader- RuntimeError: stack expects each tensor to be equal size

· 2025年4月5日 · 0 Comment

我正在尝试从头开始实现ResNet。 在将数据集设置到…

IT技术

‘Sequential’对象没有属性’classifier’

· 2025年4月5日 · 0 Comment

如何查找Pytorch模型的in_features？m…