diff --git a/nerfstudio/data/pixel_samplers.py b/nerfstudio/data/pixel_samplers.py index 6a5eff0d1..d9a181af8 100644 --- a/nerfstudio/data/pixel_samplers.py +++ b/nerfstudio/data/pixel_samplers.py @@ -69,8 +69,8 @@ class PixelSampler: else: indices = ( torch.rand((batch_size, 3), device=device) - * torch.tensor([num_images, image_height, image_width], device=device).long() - ) + * torch.tensor([num_images, image_height, image_width], device=device) + ).long() return indices