tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-12 19:04:03 +08:00
Code Activity

Add weight-decay option to RAdam optimizer (#1525)

Browse Source
This commit is contained in:
Jonáš Kulhánek authored and GitHub committed 2023-02-27 09:21:00 -08:00
1 parent 5d640bfcfd
commit 955497fcb2
1 file changed
+2
Download Patch File Download Diff File Expand all files Collapse all files
nerfstudio/engine/optimizers.py
+2
View File
@@ -66,6 +66,8 @@ class RAdamOptimizerConfig(OptimizerConfig):
"""Basic optimizer config with RAdam"""
_target: Type = torch.optim.RAdam
weight_decay: float = 0
"""The weight decay to use."""
class Optimizers: