add jt.save

2020-12-17 21:02:05 +08:00 · 2020-12-17 21:02:05 +08:00 · 2d1df7336a
parent 37d4ffca0b
commit 2d1df7336a
1 changed files with 3 additions and 0 deletions
--- a/python/jittor/init.py
+++ b/python/jittor/init.py
@ -476,6 +476,9 @@ def load(path):
        model_dict = safeunpickle(path)
    return model_dict

+def save(params_dict, path):
+    safepickle(params_dict, path)
+
 def _uniq(x):
    a = set()
    b = []