moved num samples to instant ngp config (#304)

This commit is contained in:
evonneng authored and GitHub committed 2022-08-29 15:50:59 -07:00
1 parent 04d0165a24
commit edf9f59940
2 files changed
+2 -1

No files matched your search

+1
View File
@@ -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
+1 -1
View File
@@ -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()