From 0846e45fe144168bc0d6b04ecf7a09ea5775f361 Mon Sep 17 00:00:00 2001 From: Terrance Wang Date: Wed, 18 Jan 2023 10:01:10 -0800 Subject: [PATCH] setting tensorf default from wandb to viewer (#1250) --- nerfstudio/configs/method_configs.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/nerfstudio/configs/method_configs.py b/nerfstudio/configs/method_configs.py index d11c377b5..30682a242 100644 --- a/nerfstudio/configs/method_configs.py +++ b/nerfstudio/configs/method_configs.py @@ -248,6 +248,9 @@ method_configs["vanilla-nerf"] = TrainerConfig( method_configs["tensorf"] = TrainerConfig( method_name="tensorf", + steps_per_eval_batch=500, + steps_per_save=2000, + max_num_iterations=30000, mixed_precision=False, pipeline=VanillaPipelineConfig( datamanager=VanillaDataManagerConfig( @@ -265,6 +268,8 @@ method_configs["tensorf"] = TrainerConfig( "scheduler": SchedulerConfig(lr_final=0.002, max_steps=30000), }, }, + viewer=ViewerConfig(num_rays_per_chunk=1 << 15), + vis="viewer", ) method_configs["dnerf"] = TrainerConfig(