From ce733a83081ae810ebb72fee692494e31fe180d9 Mon Sep 17 00:00:00 2001
From: Yuantao Chen <44546388+Tao-11-chen@users.noreply.github.com>
Date: Tue, 18 Apr 2023 14:35:34 +0800
Subject: [PATCH] Delete useless lists of tensors put to CUDA which may cause
 OOM. (#1748)

eval_large_imgs
---
 nerfstudio/models/base_model.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/nerfstudio/models/base_model.py b/nerfstudio/models/base_model.py
index a73ea67a1..e03893195 100644
--- a/nerfstudio/models/base_model.py
+++ b/nerfstudio/models/base_model.py
@@ -177,12 +177,12 @@ class Model(nn.Module):
             ray_bundle = camera_ray_bundle.get_row_major_sliced_ray_bundle(start_idx, end_idx)
             outputs = self.forward(ray_bundle=ray_bundle)
             for output_name, output in outputs.items():  # type: ignore
+                if not torch.is_tensor(output):
+                    # TODO: handle lists of tensors as well
+                    continue
                 outputs_lists[output_name].append(output)
         outputs = {}
         for output_name, outputs_list in outputs_lists.items():
-            if not torch.is_tensor(outputs_list[0]):
-                # TODO: handle lists of tensors as well
-                continue
             outputs[output_name] = torch.cat(outputs_list).view(image_height, image_width, -1)  # type: ignore
         return outputs