diff --git a/seathru/seathru_config.py b/seathru/seathru_config.py index c04b6ef..27837d5 100644 --- a/seathru/seathru_config.py +++ b/seathru/seathru_config.py @@ -43,8 +43,8 @@ "mode": "off", "optimizer": AdamOptimizerConfig(lr=6e-4, eps=1e-8, weight_decay=1e-2), "scheduler": ExponentialDecaySchedulerConfig( - lr_final=6e-6, max_steps=500000 - ), + lr_final=6e-6, max_steps=500000 + ), }, }, viewer=ViewerConfig(num_rays_per_chunk=1 << 15), @@ -112,8 +112,8 @@ "mode": "off", "optimizer": AdamOptimizerConfig(lr=6e-4, eps=1e-8, weight_decay=1e-2), "scheduler": ExponentialDecaySchedulerConfig( - lr_final=6e-6, max_steps=500000 - ), + lr_final=6e-6, max_steps=500000 + ), }, }, viewer=ViewerConfig(num_rays_per_chunk=1 << 15),