mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-24 04:59:47 +00:00
nit
This commit is contained in:
@@ -81,7 +81,7 @@ class AdamWConfig(OptimizerConfig):
|
|||||||
eps: float = 1e-8
|
eps: float = 1e-8
|
||||||
weight_decay: float = 1e-2
|
weight_decay: float = 1e-2
|
||||||
grad_clip_norm: float = 10.0
|
grad_clip_norm: float = 10.0
|
||||||
fused: bool = False
|
fused: bool = False
|
||||||
|
|
||||||
def build(self, params: dict) -> torch.optim.Optimizer:
|
def build(self, params: dict) -> torch.optim.Optimizer:
|
||||||
kwargs = asdict(self)
|
kwargs = asdict(self)
|
||||||
|
|||||||
@@ -79,7 +79,7 @@ class SmolVLAConfig(PreTrainedConfig):
|
|||||||
optimizer_eps: float = 1e-8
|
optimizer_eps: float = 1e-8
|
||||||
optimizer_weight_decay: float = 1e-10
|
optimizer_weight_decay: float = 1e-10
|
||||||
optimizer_grad_clip_norm: float = 10
|
optimizer_grad_clip_norm: float = 10
|
||||||
optimizer_fused: bool = False
|
optimizer_fused: bool = False
|
||||||
|
|
||||||
scheduler_warmup_steps: int = 1_000
|
scheduler_warmup_steps: int = 1_000
|
||||||
scheduler_decay_steps: int = 30_000
|
scheduler_decay_steps: int = 30_000
|
||||||
|
|||||||
Reference in New Issue
Block a user