From 3ba028785482e9dcd9534ab57cb750bbf97fbee9 Mon Sep 17 00:00:00 2001 From: Andrea Simonelli Date: Thu, 8 Aug 2024 11:00:44 +0200 Subject: [PATCH] Fix typo in examples/simple_viewer.py --- examples/simple_viewer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/simple_viewer.py b/examples/simple_viewer.py index 7a8f5b90..5b2dbe0f 100644 --- 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"]))