mipnerf in parallel

This commit is contained in:
maturk committed 2023-08-07 20:48:51 +03:00
1 parent f0574ccdbc
commit 818c0577c2
1 file changed
+1 -1
+1 -1
View File
@@ -305,7 +305,7 @@ method_configs["instant-ngp-bounded"] = TrainerConfig(
method_configs["mipnerf"] = TrainerConfig(
method_name="mipnerf",
pipeline=VanillaPipelineConfig(
datamanager=VanillaDataManagerConfig(dataparser=NerfstudioDataParserConfig(), train_num_rays_per_batch=1024),
datamanager=ParallelDataManagerConfig(dataparser=NerfstudioDataParserConfig(), train_num_rays_per_batch=1024),
model=VanillaModelConfig(
_target=MipNerfModel,
loss_coefficients={"rgb_loss_coarse": 0.1, "rgb_loss_fine": 1.0},