diff --git a/tools/modelwriter.h b/tools/modelwriter.h index 0cdb76877..107d83e05 100644 --- a/tools/modelwriter.h +++ b/tools/modelwriter.h @@ -1277,6 +1277,8 @@ int ModelWriter::save(const char* parampath, const char* binpath) fprintf_param_value(" 2=%e", width_scale) fprintf_param_value(" 3=%d", output_height) fprintf_param_value(" 4=%d", output_width) + fprintf_param_value(" 5=%d", dynamic_target_size) + fprintf_param_value(" 6=%d", align_corner) } else if (layer->type == "Log") {