diff --git a/nerfstudio/configs/method_configs.py b/nerfstudio/configs/method_configs.py index 0bc56f6da..0357d0f9b 100644 --- a/nerfstudio/configs/method_configs.py +++ b/nerfstudio/configs/method_configs.py @@ -112,6 +112,46 @@ method_configs["nerfacto"] = TrainerConfig( viewer=ViewerConfig(num_rays_per_chunk=1 << 15), vis="viewer", ) +method_configs["nerfacto-big"] = TrainerConfig( + method_name="nerfacto", + steps_per_eval_batch=500, + steps_per_save=2000, + max_num_iterations=100000, + mixed_precision=True, + pipeline=VanillaPipelineConfig( + datamanager=VanillaDataManagerConfig( + dataparser=NerfstudioDataParserConfig(), + train_num_rays_per_batch=4096, + eval_num_rays_per_batch=4096, + camera_optimizer=CameraOptimizerConfig( + mode="SO3xR3", optimizer=RAdamOptimizerConfig(lr=6e-4, eps=1e-8, weight_decay=1e-3) + ), + ), + model=NerfactoModelConfig( + eval_num_rays_per_chunk=1 << 15, + num_nerf_samples_per_ray=128, + num_proposal_samples_per_ray=(512, 256), + hidden_dim=128, + hidden_dim_color=128, + hidden_dim_transient=128, + max_res=3000, + proposal_weights_anneal_max_num_iters=5000, + log2_hashmap_size=21, + ), + ), + optimizers={ + "proposal_networks": { + "optimizer": RAdamOptimizerConfig(lr=1e-2, eps=1e-15), + "scheduler": None, + }, + "fields": { + "optimizer": RAdamOptimizerConfig(lr=1e-2, eps=1e-15), + "scheduler": ExponentialDecaySchedulerConfig(lr_final=1e-4, max_steps=100000), + }, + }, + viewer=ViewerConfig(num_rays_per_chunk=1 << 15), + vis="viewer", +) method_configs["depth-nerfacto"] = TrainerConfig( method_name="depth-nerfacto", diff --git a/tests/test_train.py b/tests/test_train.py index b38e84205..f795f6060 100644 --- a/tests/test_train.py +++ b/tests/test_train.py @@ -21,6 +21,7 @@ BLACKLIST = [ "instant-ngp", "instant-ngp-bounded", "nerfacto", + "nerfacto-big", "volinga", "phototourism", "depth-nerfacto",