refine

PaddlePaddle · Sep 20, 2024 · 065e3d8 · 065e3d8
1 parent 16189af
commit 065e3d8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tools/model_converter/convert_paddle_to_standard.py b/tools/model_converter/convert_paddle_to_standard.py
@@ -173,6 +173,7 @@ def convert_model(paddle_model, save_dir, params2val_dict):
  graph.forward_block_idx = block.forward_block_idx
 
  model_save_path = os.path.join(save_dir, "standard_model.model")
+ print("new_model: ", new_model, flush=1)
  model_str = new_model.SerializeToString()
  with open(model_save_path, "wb") as writable:
  writable.write(model_str)
@@ -184,5 +185,7 @@ def convert_model(paddle_model, save_dir, params2val_dict):
  paddle.set_device("cpu")
  if not os.path.exists(args.save_dir):
  os.mkdir(args.save_dir)
- params2val_dict, _ = convert_params(args.paddle_model, args.save_dir)
+ # params2val_dict, _ = convert_params(args.paddle_model, args.save_dir)
+ params2val_dict = {}
  convert_model(args.paddle_model, args.save_dir, params2val_dict)
+