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

explain why n_encoder_layers=1

This commit is contained in:
Alexander Soare 2024-05-17 10:22:23 +01:00
parent 4d7d41cdee
commit 40ed37417f
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -115,6 +115,9 @@ 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
# Note: Although the original ACT implementation has 7 for `n_decoder_layers`, there is a bug in the code
# that means only the first layer is used. Here we match the original implementation by setting this to 1.
# See this issue https://github.com/tonyzhaozh/act/issues/25#issue-2258740521.
n_decoder_layers: int = 1 n_decoder_layers: int = 1
# VAE. # VAE.
use_vae: bool = True use_vae: bool = True

3
lerobot/configs/policy/act.yaml
View File

@ -66,6 +66,9 @@ policy:
dim_feedforward: 3200 dim_feedforward: 3200
feedforward_activation: relu feedforward_activation: relu
n_encoder_layers: 4 n_encoder_layers: 4
# Note: Although the original ACT implementation has 7 for `n_decoder_layers`, there is a bug in the code
# that means only the first layer is used. Here we match the original implementation by setting this to 1.
# See this issue https://github.com/tonyzhaozh/act/issues/25#issue-2258740521.
n_decoder_layers: 1 n_decoder_layers: 1
# VAE. # VAE.
use_vae: true use_vae: true