MindSpore ResNet 模型训练:解决 'ValueError: For 'Conv2D', the 'pad' must be zero when 'pad_mode' is not "pad", but got 'pad' and 'pad_mode' is same.' 错误
这个错误是因为在 ResNet 模型的定义中,有一些卷积层的 padding 和 pad_mode 参数设置有误。默认的 padding 是 'valid',即不进行 padding,但是 pad_mode 参数被设置为了 'same',这导致了错误的发生。解决方法是将这些卷积层的 padding 设置为 0,或将 pad_mode 设置为 'pad'。例如,在 ResNet 类的 init 方法中,将第一个卷积层的定义修改为:
self.conv1 = nn.Conv2d(1, 64, kernel_size=7, stride=2, padding=0, has_bias=False)
同样地,将其他卷积层的定义中的 padding 或 pad_mode 参数进行相应的修改即可。
原文地址: https://www.cveoy.top/t/topic/mQjv 著作权归作者所有。请勿转载和采集!