From f228299601e372c6a6739721542d0ed3a0c21656 Mon Sep 17 00:00:00 2001
From: tangwei <tangwei12@baidu.com>
Date: Mon, 27 Apr 2020 15:31:17 +0800
Subject: [PATCH] add save model to proto  util

---
 fleetrec/core/utils/util.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/fleetrec/core/utils/util.py b/fleetrec/core/utils/util.py
index 6c6bd12a..b410df1d 100755
--- a/fleetrec/core/utils/util.py
+++ b/fleetrec/core/utils/util.py
@@ -16,9 +16,20 @@ import os
 import time
 import datetime
 
+from paddle import fluid
 from fleetrec.core.utils import fs as fs
 
 
+def save_program_proto(path, program=None):
+    if program is None:
+        _program = fluid.default_main_program()
+    else:
+        _program = program
+
+    with open(path, "wb") as f:
+        f.write(_program.desc.serialize_to_string())
+
+
 def str2bool(v):
     if isinstance(v, bool):
         return v
-- 
GitLab