Abdulrahman1989 commited on
Commit
91765b3
·
1 Parent(s): aafb120

fix self device

Browse files
Files changed (1) hide show
  1. Image3DProcessor.py +3 -3
Image3DProcessor.py CHANGED
@@ -96,9 +96,9 @@ class Image3DProcessor:
96
  for r_idx in range(world_view_transforms.shape[0]):
97
  rendered_image = render_predicted(
98
  reconstruction,
99
- world_view_transforms[r_idx].to(device),
100
- full_proj_transforms[r_idx].to(device),
101
- camera_centers[r_idx].to(device),
102
  background,
103
  model_cfg,
104
  focals_pixels=None
 
96
  for r_idx in range(world_view_transforms.shape[0]):
97
  rendered_image = render_predicted(
98
  reconstruction,
99
+ world_view_transforms[r_idx].to(self.device),
100
+ full_proj_transforms[r_idx].to(self.device),
101
+ camera_centers[r_idx].to(self.device),
102
  background,
103
  model_cfg,
104
  focals_pixels=None