diff --git a/lerobot/common/policies/act/configuration_act.py b/lerobot/common/policies/act/configuration_act.py index be444b06..ab0f95c0 100644 --- 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