From bd1060ee83acf9c9ebd9b39197badbd9af9466df Mon Sep 17 00:00:00 2001
From: hturki <turki.haithem@gmail.com>
Date: Mon, 16 Jan 2023 18:21:44 -0500
Subject: [PATCH] save image_coords as non-persistent buffer (#1216)

Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
---
 nerfstudio/model_components/ray_generators.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nerfstudio/model_components/ray_generators.py b/nerfstudio/model_components/ray_generators.py
index 87a30cf67..a894da854 100644
--- a/nerfstudio/model_components/ray_generators.py
+++ b/nerfstudio/model_components/ray_generators.py
@@ -36,7 +36,7 @@ class RayGenerator(nn.Module):
         super().__init__()
         self.cameras = cameras
         self.pose_optimizer = pose_optimizer
-        self.image_coords = nn.Parameter(cameras.get_image_coords(), requires_grad=False)
+        self.register_buffer("image_coords", cameras.get_image_coords(), persistent=False)
 
     def forward(self, ray_indices: TensorType["num_rays", 3]) -> RayBundle:
         """Index into the cameras to generate the rays.