diff --git a/lerobot/configs/policy/act_aloha_real.yaml b/lerobot/configs/policy/act_aloha_real.yaml index 70da3c56..7c8094da 100644 --- a/lerobot/configs/policy/act_aloha_real.yaml +++ b/lerobot/configs/policy/act_aloha_real.yaml @@ -114,7 +114,7 @@ policy: n_vae_encoder_layers: 4 # Inference. - temporal_ensemble_momentum: null + temporal_ensemble_coeff: null # Training and loss computation. dropout: 0.1 diff --git a/lerobot/configs/policy/act_koch_real.yaml b/lerobot/configs/policy/act_koch_real.yaml index fd4bf3b5..6ddebab1 100644 --- a/lerobot/configs/policy/act_koch_real.yaml +++ b/lerobot/configs/policy/act_koch_real.yaml @@ -95,7 +95,7 @@ policy: n_vae_encoder_layers: 4 # Inference. - temporal_ensemble_momentum: null + temporal_ensemble_coeff: null # Training and loss computation. dropout: 0.1 diff --git a/lerobot/configs/policy/act_moss_real.yaml b/lerobot/configs/policy/act_moss_real.yaml index 840a64e1..d996c359 100644 --- a/lerobot/configs/policy/act_moss_real.yaml +++ b/lerobot/configs/policy/act_moss_real.yaml @@ -95,7 +95,7 @@ policy: n_vae_encoder_layers: 4 # Inference. - temporal_ensemble_momentum: null + temporal_ensemble_coeff: null # Training and loss computation. dropout: 0.1 diff --git a/lerobot/configs/policy/act_so100_real.yaml b/lerobot/configs/policy/act_so100_real.yaml index 3a0975b6..cf5b1f14 100644 --- a/lerobot/configs/policy/act_so100_real.yaml +++ b/lerobot/configs/policy/act_so100_real.yaml @@ -95,7 +95,7 @@ policy: n_vae_encoder_layers: 4 # Inference. - temporal_ensemble_momentum: null + temporal_ensemble_coeff: null # Training and loss computation. dropout: 0.1