This commit is contained in:
Justin committed 2023-06-21 22:26:27 +00:00
1 parent 1d8cef26f7
commit 3f73e78bfd
1 file changed
+2 -2
+2 -2
View File
@@ -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