From 77634b97cf4ea921c71403416d9d3a51c41aea76 Mon Sep 17 00:00:00 2001 From: terrance Date: Fri, 3 Feb 2023 22:40:32 +0000 Subject: [PATCH] add proposal network optimizer to config --- nerfstudio/configs/method_configs.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/nerfstudio/configs/method_configs.py b/nerfstudio/configs/method_configs.py index fee855fac..0c1b78dcf 100644 --- a/nerfstudio/configs/method_configs.py +++ b/nerfstudio/configs/method_configs.py @@ -352,6 +352,10 @@ method_configs["dreamfusion"] = DreamfusionTrainerConfig( model=DreamFusionModelConfig(), ), optimizers={ + "proposal_networks": { + "optimizer": AdamOptimizerConfig(lr=1e-2, eps=1e-15), + "scheduler": None, + }, "fields": { "optimizer": AdamOptimizerConfig(lr=5e-3, eps=1e-15), "scheduler": None,