torchvision | 开发者交流平台

如何在PyTorch中创建一个高效的条件层？

英文: How to create an efficient conditional layer in pytorch? 问题我有一个ResNet50模型，它输出一个类别预测（1、2或3）。基于分类...

2023年5月29日110评论

英文: Error: ImportError: cannot import name 'model_urls' from 'torchvision.models.vgg'...

2023年5月17日160评论

英文: How to retrieve Dataset object from DataLoader? 问题 I have a PyTorch DataLoader and want to retri...

2023年3月9日178评论

英文: How to make shifts in tensors (images) without having to make loops? 问题我想对张量（图像集合）执行位移操作，例如使用to...

2023年2月27日168评论

英文: model with CrossEntropyLoss criterion doesnt apply softmax pytorch 问题我正在开发一个模型，使用nn.CrossEntrop...

2023年2月18日115评论

英文: Apply Torchvision Transforms to a numpy array 问题我编写了以下代码： transform = transforms.Compose([ tran...

2023年2月18日107评论

英文: why pytorch is not installing on windows 11 on python 3.11 问题我正在尝试在Windows 11上安装PyTorch已经有一段时间了...

2023年2月9日155评论