提交 b5c92092 编写于 作者: C chengduoZH

fix conv_op bug

上级 4948f7b3
......@@ -55,7 +55,7 @@ void ConvOp::InferShape(framework::InferShapeContext* ctx) const {
std::vector<int64_t> output_shape({in_dims[0], filter_dims[0]});
for (size_t i = 0; i < strides.size(); ++i) {
PADDLE_ENFORCE(in_dims[i + 2] + 2 * paddings[i] -
(dilations[i] * (filter_dims[i + 2] - 1) + 1) >
(dilations[i] * (filter_dims[i + 2] - 1) + 1) >=
0,
"Due to the settings of paddings, filter_dims and "
"dilations, the output size is less than 0, please check "
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册