Fix typo in examples/simple_viewer.py (#321)

nerfstudio-project · Aug 8, 2024 · 8bd343f · 8bd343f
1 parent a6c2b52
commit 8bd343f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/simple_viewer.py b/examples/simple_viewer.py
@@ -109,7 +109,7 @@ def main(local_rank: int, world_rank, world_size: int, args):
         means, quats, scales, opacities, sh0, shN = [], [], [], [], [], []
         for ckpt_path in args.ckpt:
             ckpt = torch.load(ckpt_path, map_location=device)["splats"]
-            means.append(ckpt["means3d"])
+            means.append(ckpt["means"])
             quats.append(F.normalize(ckpt["quats"], p=2, dim=-1))
             scales.append(torch.exp(ckpt["scales"]))
             opacities.append(torch.sigmoid(ckpt["opacities"]))