diff --git a/nerfactory/configs/base.py b/nerfactory/configs/base.py index 1ce43232b..d4b84eef4 100644 --- a/nerfactory/configs/base.py +++ b/nerfactory/configs/base.py @@ -251,6 +251,7 @@ class InstantNGPModelConfig(ModelConfig): enable_collider: bool = False field_implementation: Literal["torch", "tcnn"] = "tcnn" # torch, tcnn, ... loss_coefficients: Dict[str, float] = to_immutable_dict({"rgb_loss": 1.0}) + num_samples: int = 1024 # instead of course/fine samples @dataclass diff --git a/nerfactory/models/instant_ngp.py b/nerfactory/models/instant_ngp.py index c12823d4d..afbbf93e2 100644 --- a/nerfactory/models/instant_ngp.py +++ b/nerfactory/models/instant_ngp.py @@ -68,7 +68,7 @@ class NGPModel(Model): def populate_misc_modules(self): # samplers - self.sampler = NGPSpacedSampler(num_samples=1024, density_field=self.density_field) + self.sampler = NGPSpacedSampler(num_samples=self.config.num_samples, density_field=self.density_field) # losses self.rgb_loss = MSELoss()