add proposal network optimizer to config

This commit is contained in:
terrance committed 2023-02-03 22:40:32 +00:00
1 parent 3dffd956fe
commit 77634b97cf
1 file changed
+4
+4
View File
@@ -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,