update

common.py

@@ -179,6 +179,8 @@ def render_video(
     render_fn = (
         render_mesh if isinstance(sample, MeshExtractResult) else render_frames
     )
+    print(torch.cuda.memory_allocated() / 1024**2, "MB 已分配")
+    print(torch.cuda.memory_reserved() / 1024**2, "MB 已预留")
     result = render_fn(
         sample,
         extrinsics,
@@ -186,6 +188,8 @@ def render_video(
         {"resolution": resolution, "bg_color": bg_color},
         **kwargs,
     )
+    print(torch.cuda.memory_allocated() / 1024**2, "MB 已分配")
+    print(torch.cuda.memory_reserved() / 1024**2, "MB 已预留")
 
     return result