mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-13 03:14:00 +08:00
moved num samples to instant ngp config (#304)
This commit is contained in:
1 parent
04d0165a24
commit
edf9f59940
2 files changed
+2
-1
No files matched your search
@@ -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
|
||||
|
||||
@@ -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()
|
||||
|
||||
Reference in new issue
Block a user