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