From 3f73e78bfd354066e8821b8340984f55b2b90f65 Mon Sep 17 00:00:00 2001 From: Justin Date: Wed, 21 Jun 2023 22:26:27 +0000 Subject: [PATCH] oops --- nerfstudio/data/pixel_samplers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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