Skip to content

Commit b5c9209

Browse files
committed
fix conv_op bug
1 parent 4948f7b commit b5c9209

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/fluid/operators/conv_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ void ConvOp::InferShape(framework::InferShapeContext* ctx) const {
5555
std::vector<int64_t> output_shape({in_dims[0], filter_dims[0]});
5656
for (size_t i = 0; i < strides.size(); ++i) {
5757
PADDLE_ENFORCE(in_dims[i + 2] + 2 * paddings[i] -
58-
(dilations[i] * (filter_dims[i + 2] - 1) + 1) >
58+
(dilations[i] * (filter_dims[i + 2] - 1) + 1) >=
5959
0,
6060
"Due to the settings of paddings, filter_dims and "
6161
"dilations, the output size is less than 0, please check "

0 commit comments

Comments
 (0)