add ONEFLOW_DEBUG_CHECKPOINT

Oneflow-Inc · Aug 8, 2023 · 1757891 · 1757891
1 parent b664acc
commit 1757891
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/oneflow/framework/check_point_v2.py b/python/oneflow/framework/check_point_v2.py
@@ -527,6 +527,9 @@ def is_file_and_support_pytorch_format(
             content = torch.load(path, map_location="cpu")
             return True, (content,)
     except:
+        if os.getenv("ONEFLOW_DEBUG_CHECKPOINT") == "1":
+            import traceback
+            traceback.print_exc()
         return False