pytorch 相关问题

PyTorch是一个深度学习框架，它实现了一个动态计算图，它允许您改变神经网络在运行中的行为方式，并能够执行向后自动区分。

有没有办法将 pytorch DataLoader (torch.utils.data.Dataloader) 完全加载到我的 GPU 中？现在，我将每个批次分别加载到我的 GPU 中。 CTX = torch.device('cuda') train_loader = torch.util...

python pytorch gpu dataloader

回答 2 投票 0

如何为 PyTorch TensorBoard 将多条 PR 曲线绘制到一张图表上？

下面是绘制PR曲线的代码取自https://pytorch.org/tutorials/intermediate/tensorboard_tutorial.html#assessing-trained-models-with-tensorboard 从 torch.utils.tensorboard 导入

python pytorch tensorboard

回答 1 投票 0

没有名为“torch._six”的模块可导入队列

我正在尝试应用来自 github https://github.com/sanghyun-son/EDSR-PyTorch 的 RCAN 模型。我正在使用 pytorch 2.0.1。我收到错误： ModuleNotFoundError：没有名为“torch._six”的模块 ...

pytorch pytorch-dataloader

回答 1 投票 0

无法训练我的 UNET 多类分割模型

我尝试用 pytorch 从头开始制作我的 UNET，但什么也没得到，但黑色蒙版作为我的模型的输出。我需要对汽车的损坏进行分段，因此我实现了彩色贴图。我确信 70% 的事情是......

machine-learning deep-learning pytorch image-segmentation

回答 1 投票 0

如何修复调用 `cublasSgemm` 时的 RuntimeError CUDA 错误 CUBLAS_STATUS_INVALID_VALUE？

在工作的 cuda 环境上训练某些模型时，调用 cublasSgemm(handle, opa, opb, m, n, k, &alpha,... 时会出现错误 RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE

pytorch

回答 3 投票 0

如何可视化带边缘的分段任意模型？

我可以使用分段任意模型并可视化其结果，但它们看起来与在线演示不同。我的结果看起来像这样：这是该部分的结果 -

python pytorch computer-vision image-segmentation

回答 1 投票 0

Pytorch + Ray Tune 报告 ImplicitFunc 太大，不知道哪个引用大

与这个问题类似，Ray Tune 正在向我报告： ValueError：参与者 ImplicitFunc 太大 (421 MiB > FUNCTION_SIZE_ERROR_THRESHOLD=95 MiB)。检查它的定义是否隐式...

python machine-learning pytorch cross-validation ray

回答 1 投票 0

参数和子项有什么区别？

看起来parameters和children显示的信息是一样的，那么它们之间有什么区别呢？进口火炬 print('火炬.__版本__', 火炬.__版本__) m = torch.load('imagenet_resnet18.pth...

python pytorch

回答 2 投票 0

计算滑动“图像相似度”分数，类似于 conv2d

我有一个大图像和一个较小的“内核”图像。我想将内核与图像的每个部分进行比较（通过在图像上“滑动”内核）并检索“

pytorch convolution

回答 1 投票 0

我尝试使用pytorchchildren()将resnet分成两部分，但它不起作用

这是一个简单的例子。我尝试使用子节点将网络（Resnet50）分为两部分：头部和尾部。从概念上讲，这应该可行，但事实并非如此。这是为什么呢？进口火炬导入 torch.nn...

python pytorch

回答 2 投票 0

PyTorch：计算类似于 conv2d 的滑动“图像相似度”分数

我有一个大图像和一个较小的“内核”图像。我想将内核与图像的每个部分进行比较（通过在图像上“滑动”内核）并检索“

pytorch convolution

回答 1 投票 0

在 PyTorch 中，如何计算矩阵乘法相对于前向传播内隐藏状态的梯度？

这是我正在研究的模型的简化版本：类 InferContextModel(nn.Module): def __init__(自身，输入大小，上下文大小，输出大小)：超级().__init__() 自我。

pytorch gradient

回答 1 投票 0

用户警告：针对 CuDNN 问题应用了解决方法，安装 nvrtc.so

我正在下面训练一个卷积神经网络。 def __init__(self, n_channels, n_classes): 超级().__init__() self.model = models.mobilenet_v3_large(预训练=True) self.model.class...

pytorch conv-neural-network

回答 2 投票 0

我正在尝试使用 Pytorch 定义 NN 模型，但是当我想将 y-train 转换为 y-train-tensor 时出现类型错误，我应该如何修复它？

这是我的代码。我将威斯康星州乳腺癌数据库从 kaggle 导入到 vscode。所有数据集变量均为数字和浮点 64。存在此类型错误（发生异常：TypeError 不能骗...

pytorch neural-network typeerror numpy-ndarray

回答 1 投票 0

导出Detectron2模型

我正在尝试从使用 detectorron2 的 panoptic-deeplab 项目中导出模型。我想将其导出为 .pt，以便稍后可以将其加载到 LibTorch 中。我想预测全景分割...

pytorch export jit detectron

回答 1 投票 0

AttributeError：“元组”没有属性“to”

我正在编写这个图像分类器，并且已经定义了加载器，但是遇到了这个错误，我对此一无所知。我已经定义了火车装载机，为了更好的解释我尝试了这个对于...

machine-learning deep-learning pytorch kaggle

回答 2 投票 0

问题：torch.distributed.barrier() 卡住了。怎么解决？

我正在尝试使用 DistributedDataParallel 并已为其编写代码。但是，我遇到了一个问题，当我在 init_process 之后使用 torch.distributed.barrier() 时，进程会停止...

python pytorch

回答 1 投票 0

如何修复错误 ValueError: Expected input batch_size (49) to match target batch_size (64)

我一直在修改数据加载器端，但它仍然显示该错误。这是预处理代码：进口火炬从 torch.utils.data 导入数据集，DataLoader 从 torchvision 导入转换...

python machine-learning deep-learning pytorch neural-network

回答 1 投票 0

我应该安装哪个版本的torch和torchtext

我使用的是Windows，Python版本为3.11.4，pandas版本为2.2.1 我正在尝试安装 torch 和 torchtext 但总是出现依赖错误。安装的正确版本是什么...

python pytorch torch

回答 1 投票 0

Pytorch 模型对象没有属性“预测”BERT

我使用 pytorch 训练了 BertClassifier 模型。创建我的 best.pt 之后，我想在生产中制作我的模型并使用它从样本开始进行预测和分类，所以我恢复它们

python pytorch huggingface-transformers bert-language-model sentence-transformers

回答 1 投票 0

pytorch 相关问题

最新问题