update None check for ray_directions_ablation

replace it with `torch.is_tensor`
gafniguy · Sep 14, 2022 · 18e83b2 · 18e83b2
1 parent 951c177
commit 18e83b2
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/nerface_code/nerf-pytorch/nerf/train_utils.py b/nerface_code/nerf-pytorch/nerf/train_utils.py
@@ -179,9 +179,7 @@ def run_one_iter_of_nerf(
     latent_code = None,
     ray_directions_ablation = None
 ):
-    is_rad = False
-    if ray_directions_ablation:
-        is_rad = True
+    is_rad = torch.is_tensor(ray_directions_ablation)
     viewdirs = None
     if options.nerf.use_viewdirs:
         # Provide ray directions as input