mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-13 19:33:46 +08:00
oops
This commit is contained in:
1 parent
1d8cef26f7
commit
3f73e78bfd
1 file changed
+2
-2
@@ -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
|
||||
|
||||
|
||||
Reference in new issue
Block a user