MindSpore报错: Conv2D输入shape填充后的高和宽必须大于卷积核的高和宽

1 报错描述

1.1 系统环境

Hardware Environment(Ascend/GPU/CPU): Ascend Software Environment: -- MindSpore version (source or binary): 2.0.0 -- Python version (e.g., Python 3.7.5): 3.7.6 -- OS platform and distribution (e.g., Linux Ubuntu 16.04): Ubuntu 4.15.0-74-generic -- GCC/Compiler version (if compiled from source):

1.2 基本信息

1.2.1 脚本

训练脚本是通过构建Conv2D的单算子网络,来实现2d卷积操作的。脚本如下:

 01 class Net(nn.Cell):
 02     def __init__(self, in_channels, out_channels, kernel_size):
 03         super(Net, self).__init__()
 04         self.in_channels = in_channels
 05         self.out_channels = out_channels
 06         self.kernel_size = kernel_size
 07         # self.pad_mode = "valid"
 08         self.conv2d = nn.Conv2d(self.in_channels, self.out_channels, self.kernel_size, pad_mode="valid")
 09 
 10     def construct(self, x):
 11      

猜你喜欢

转载自blog.csdn.net/beauty0220/article/details/129139287