in shap_e/models/nerf/model.py [0:0]
def __init__(
self,
# Positional encoding parameters
n_levels: int = 10,
# MLP parameters
d_hidden: int = 256,
n_density_layers: int = 4,
n_channel_layers: int = 1,
n_channels: int = 3,
sh_degree: int = 4,
activation: str = "relu",
density_activation: str = "exp",
init: Optional[str] = None,
init_scale: float = 1.0,
output_activation: str = "sigmoid",
meta_parameters: bool = False,
trainable_meta: bool = False,
zero_out: bool = True,
register_freqs: bool = True,
posenc_version: str = "v1",
device: torch.device = torch.device("cuda"),