From 9fe843fa398dee90c1fb2aeb24ddc8da184830d2 Mon Sep 17 00:00:00 2001 From: Chen Xin Date: Fri, 21 Apr 2023 17:31:01 +0800 Subject: [PATCH] fix wrong param id of Squeeze for modelwriter (#4658) --- tools/modelwriter.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/modelwriter.h b/tools/modelwriter.h index 9103e3004..8c74d5946 100644 --- a/tools/modelwriter.h +++ b/tools/modelwriter.h @@ -2307,7 +2307,7 @@ int ModelWriter::save(const char* parampath, const char* binpath) fprintf_param_value(" 11=%d", squeeze_d) fprintf_param_value(" 2=%d", squeeze_c) { - if (!op->axes.empty()) fprintf_param_int_array(0, op->axes, pp); + if (!op->axes.empty()) fprintf_param_int_array(3, op->axes, pp); } } else if (layer->type == "Threshold")