From 128f45a3c1d2f0e6d30008428d5b90eac10a4bed Mon Sep 17 00:00:00 2001 From: zhoulei20 Date: Thu, 27 Feb 2025 16:06:46 +0800 Subject: [PATCH] fix conv2d_transpose op --- paddle2onnx/mapper/nn/conv2d_transpose.h | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/paddle2onnx/mapper/nn/conv2d_transpose.h b/paddle2onnx/mapper/nn/conv2d_transpose.h index 3b4960706..b44fbb7ef 100755 --- a/paddle2onnx/mapper/nn/conv2d_transpose.h +++ b/paddle2onnx/mapper/nn/conv2d_transpose.h @@ -29,13 +29,19 @@ class Conv2dTransposeMapper : public Mapper { GetAttr("dilations", &dilations_); GetAttr("strides", &strides_); GetAttr("paddings", &paddings_); - GetAttr("padding_algorithm", &padding_algorithm_); + if (HasAttr("padding_algorithm")) { + GetAttr("padding_algorithm", &padding_algorithm_); + } else { + padding_algorithm_ = "EXPLICIT"; + } GetAttr("data_format", &data_format_); if (HasAttr("output_padding")) { GetAttr("output_padding", &output_padding_); } - GetAttr("output_size", &output_size_); + if (HasAttr("output_size")) { + GetAttr("output_size", &output_size_); + } if (paddings_.size() == 2) { paddings_.push_back(paddings_[0]); paddings_.push_back(paddings_[1]);