diff --git a/projects/nerf/nerf/implicit_function.py b/projects/nerf/nerf/implicit_function.py index ef1278bb..4209e53c 100644 --- a/projects/nerf/nerf/implicit_function.py +++ b/projects/nerf/nerf/implicit_function.py @@ -26,7 +26,7 @@ class NeuralRadianceField(torch.nn.Module): n_hidden_neurons_xyz: int = 256, n_hidden_neurons_dir: int = 128, n_layers_xyz: int = 8, - append_xyz: Tuple[int] = (5,), + append_xyz: Tuple[int, ...] = (5,), use_multiple_streams: bool = True, **kwargs, ): @@ -253,7 +253,7 @@ class MLPWithInputSkips(torch.nn.Module): output_dim: int, skip_dim: int, hidden_dim: int, - input_skips: Tuple[int] = (), + input_skips: Tuple[int, ...] = (), ): """ Args: diff --git a/projects/nerf/nerf/nerf_renderer.py b/projects/nerf/nerf/nerf_renderer.py index 8f84b0ad..d7208973 100644 --- a/projects/nerf/nerf/nerf_renderer.py +++ b/projects/nerf/nerf/nerf_renderer.py @@ -67,7 +67,7 @@ class RadianceFieldRenderer(torch.nn.Module): n_hidden_neurons_xyz: int = 256, n_hidden_neurons_dir: int = 128, n_layers_xyz: int = 8, - append_xyz: Tuple[int] = (5,), + append_xyz: Tuple[int, ...] = (5,), density_noise_std: float = 0.0, visualization: bool = False, ):