diff --git a/mace/ops/conv_pool_2d_base.h b/mace/ops/conv_pool_2d_base.h index 96b544d7bdeb52e4a70146873009fee5b0656fdb..9b1838a36c2ef4f68217616569edcee32d3a6f9e 100644 --- a/mace/ops/conv_pool_2d_base.h +++ b/mace/ops/conv_pool_2d_base.h @@ -18,7 +18,7 @@ class ConvPool2dOpBase : public Operator { strides_(OperatorBase::GetRepeatedArgument("strides")), padding_(static_cast(OperatorBase::GetSingleArgument( "padding", static_cast(SAME)))), - dilations_(OperatorBase::GetRepeatedArgument("dilations")) {} + dilations_(OperatorBase::GetRepeatedArgument("dilations", {1, 1})) {} void CalOutputSize(const index_t *input_shape, // NCHW const index_t *filter_shape, // OIHW