wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

set n_encoder_layers for ACT to 7

This commit is contained in:
Radek Osmulski 2024-05-17 18:07:01 +10:00
parent 4d7d41cdee
commit d2ae384b8a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/policies/act/configuration_act.py
View File

@ -115,7 +115,7 @@ class ACTConfig:
dim_feedforward: int = 3200 dim_feedforward: int = 3200
feedforward_activation: str = "relu" feedforward_activation: str = "relu"
n_encoder_layers: int = 4 n_encoder_layers: int = 4
n_decoder_layers: int = 1 n_decoder_layers: int = 7
# VAE. # VAE.
use_vae: bool = True use_vae: bool = True
latent_dim: int = 32 latent_dim: int = 32