Merge branch 'update-scheduler' of https://github.com/HrsPythonix/nerfstudio into update-scheduler

This commit is contained in:
Pythonix Huang committed 2023-04-01 03:09:15 +08:00
commit b31575385c
1 file changed
+2 -2
+2 -2
View File
@@ -88,9 +88,9 @@ method_configs["nerfacto"] = TrainerConfig(
train_num_rays_per_batch=4096,
eval_num_rays_per_batch=4096,
camera_optimizer=CameraOptimizerConfig(
mode="SO3xR3",
mode="SO3xR3",
optimizer=AdamOptimizerConfig(lr=6e-4, eps=1e-8, weight_decay=1e-2),
scheduler=ExponentialDecaySchedulerConfig(lr_final=6e-6, max_steps=200000),
scheduler = ExponentialDecaySchedulerConfig(lr_final=6e-6, max_steps=200000),
),
),
model=NerfactoModelConfig(eval_num_rays_per_chunk=1 << 15),