From 73664568356953f166d3cb0f3bb10da2e22accd2 Mon Sep 17 00:00:00 2001 From: Georgy Ponimatkin <38923619+ponimatkin@users.noreply.github.com> Date: Tue, 28 Feb 2023 18:22:22 +0100 Subject: [PATCH] Adding conversion to torch.float32 in get_interpolated_poses_many. (#1532) --- nerfstudio/cameras/camera_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/cameras/camera_utils.py b/nerfstudio/cameras/camera_utils.py index b1dd5a267..dae490626 100644 --- a/nerfstudio/cameras/camera_utils.py +++ b/nerfstudio/cameras/camera_utils.py @@ -229,7 +229,7 @@ def get_interpolated_poses_many( traj = np.stack(traj, axis=0) k_interp = np.stack(k_interp, axis=0) - return torch.tensor(traj), torch.tensor(k_interp) + return torch.tensor(traj, dtype=torch.float32), torch.tensor(k_interp, dtype=torch.float32) def normalize(x: torch.Tensor) -> TensorType[...]: