Skip to content

Commit d67662d

Browse files
nikhilaravifacebook-github-bot
authored andcommitted
Update use of select_cameras
Summary: Removed `select_cameras.py` from implicitron and updated all callsites to directly index the cameras. Reviewed By: bottler Differential Revision: D33187605 fbshipit-source-id: aaf5b36aef9d72db0c7e89dec519f23646f6aa05
1 parent 28ccdb7 commit d67662d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch3d/renderer/cameras.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -383,7 +383,7 @@ def __getitem__(
383383

384384
kwargs = {}
385385

386-
if not isinstance(index, (int, list, torch.LongTensor)):
386+
if not isinstance(index, (int, list, torch.LongTensor, torch.cuda.LongTensor)):
387387
msg = "Invalid index type, expected int, List[int] or torch.LongTensor; got %r"
388388
raise ValueError(msg % type(index))
389389

0 commit comments

Comments
 (0)