fix wrong param id of Squeeze for modelwriter (#4658)

Tencent · Apr 21, 2023 · 9fe843f · 9fe843f
1 parent 8adf368
commit 9fe843f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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")