diff --git a/ppgan/utils/filesystem.py b/ppgan/utils/filesystem.py
index 9b0ce88bfa5a230a06c7e32bd4d9634de6e56c71..a43c18f985bc623d31790ff4bbea172687ad4ff7 100644
--- a/ppgan/utils/filesystem.py
+++ b/ppgan/utils/filesystem.py
@@ -53,8 +53,7 @@ def save(state_dicts, file_name):
         else:
             final_dict[k] = v
 
-    with open(file_name, 'wb') as f:
-        pickle.dump(final_dict, f, protocol=2)
+    paddle.save(final_dict, file_name)
 
 
 def load(file_name):