小编Sad*_*yan的帖子

UserWarning:针对 CuDNN 问题应用了解决方法,安装 nvrtc.so

我正在下面训练一个卷积神经网络。

def __init__(self, n_channels, n_classes):
    super().__init__()
    self.model = models.mobilenet_v3_large(pretrained=True)
    self.model.classifier[-1] = nn.Linear(1280, n_classes)
    self.model.features[0][0] = nn.Conv2d(
        n_channels,
        16,
        kernel_size=(3, 3),
        stride=(2, 2),
        padding=(1, 1),
        bias=False,
    )
Run Code Online (Sandbox Code Playgroud)

警告中只有另一件事是

返回 F.conv2d(输入, 权重, 偏差, self.stride,

我不知道这个警告是什么以及我能做什么。如果需要有关系统的任何信息,我可以编辑问题并添加到其中。

conv-neural-network pytorch

6
推荐指数
1
解决办法
6533
查看次数

标签 统计

conv-neural-network ×1

pytorch ×1