From 0264aacbe16b11b2d89836b55cf4ad051c8fdfba Mon Sep 17 00:00:00 2001 From: Zak Murez <33580232+zmurez@users.noreply.github.com> Date: Tue, 22 Nov 2022 11:07:36 -0800 Subject: [PATCH] create rays on device when rendering (#993) Co-authored-by: zmurez --- scripts/render.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/scripts/render.py b/scripts/render.py index 6cca7585f..09dab4c44 100644 --- a/scripts/render.py +++ b/scripts/render.py @@ -58,6 +58,7 @@ def _render_trajectory_video( CONSOLE.print("[bold green]Creating trajectory video") images = [] cameras.rescale_output_resolution(rendered_resolution_scaling_factor) + cameras = cameras.to(pipeline.device) progress = Progress( TextColumn(":movie_camera: Rendering :movie_camera:"), @@ -71,7 +72,7 @@ def _render_trajectory_video( output_image_dir.mkdir(parents=True, exist_ok=True) with progress: for camera_idx in progress.track(range(cameras.size), description=""): - camera_ray_bundle = cameras.generate_rays(camera_indices=camera_idx).to(pipeline.device) + camera_ray_bundle = cameras.generate_rays(camera_indices=camera_idx) with torch.no_grad(): outputs = pipeline.model.get_outputs_for_camera_ray_bundle(camera_ray_bundle) render_image = []