From 4e705685101e4c1380a1be6cba9428a17d3ff286 Mon Sep 17 00:00:00 2001
From: Matt Tancik <tancik@berkeley.edu>
Date: Thu, 7 Apr 2022 17:20:01 -0700
Subject: [PATCH] update render heads

---
 mattport/nerf/render_heads.py | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)

diff --git a/mattport/nerf/render_heads.py b/mattport/nerf/render_heads.py
index 7c8de087c..7dabcade8 100644
--- a/mattport/nerf/render_heads.py
+++ b/mattport/nerf/render_heads.py
@@ -8,19 +8,17 @@ from torch import nn
 from torchtyping import TensorType
 
 
-class RenderHeadKeys(Enum):
-    """Keys for model outputs"""
-
-    DENSITY = "density"
-    DISTS = "dists"
-    RGB = "rgb"
-    SH_RGB = "sh_rgb"
-
-
 class RenderHead(nn.Module):
     """Base Render head"""
 
     def __init__(self, in_dim: int, out_dim: int, activation: Optional[nn.Module]) -> None:
+        """
+
+        Args:
+            in_dim (int): input dimension
+            out_dim (int): output dimension for renderer
+            activation (Optional[nn.Module]): output head activation
+        """
         super().__init__()
         layers = [nn.Linear(in_dim, out_dim)]
         if activation: