我正在下面训练一个卷积神经网络。
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,
我不知道这个警告是什么以及我能做什么。如果需要有关系统的任何信息,我可以编辑问题并添加到其中。