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

在Ubuntu系统上使用MindSpore 2.0.0时,遇到Conv2D算子报错:输入shape在填充后的高和宽必须大于卷积核的高和宽。报错源于Conv2dInferShape函数,原因是使用了'valid'模式且参数设置不正确。解决方案是调整输入参数或更改pad_mode为'same',确保满足卷积运算的要求。
摘要由CSDN通过智能技术生成

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      
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值