set n_encoder_layers for ACT to 7

2024-05-17 18:07:01 +10:00 · 2024-05-17 18:07:01 +10:00 · d2ae384b8a
parent 4d7d41cdee
commit d2ae384b8a
1 changed files with 1 additions and 1 deletions
--- a/lerobot/common/policies/act/configuration_act.py
+++ b/lerobot/common/policies/act/configuration_act.py
@ -115,7 +115,7 @@ class ACTConfig:
    dim_feedforward: int = 3200
    feedforward_activation: str = "relu"
    n_encoder_layers: int = 4
-    n_decoder_layers: int = 1
+    n_decoder_layers: int = 7
    # VAE.
    use_vae: bool = True
    latent_dim: int = 32