Using with stmt

AugWrite · Aug 7, 2022 · c8311a4 · c8311a4
1 parent 4f120f9
commit c8311a4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/body/human_pose/ambiguity_aware/lib/utils/misc.py b/body/human_pose/ambiguity_aware/lib/utils/misc.py
@@ -48,9 +48,8 @@ def init_weights(model):
             m.bias.data.zero_()
 
 def save_pickle(data, save_path):
-    f = open(save_path, "wb")
-    pkl.dump(data, f)
-    f.close()
+    with open(save_path, "wb") as f:
+        pkl.dump(data, f)
     print(f"=> saved to {save_path}")
 
 def load_pickle(load_path):