diff --git a/lerobot/configs/policy/act_aloha_real.yaml b/lerobot/configs/policy/act_aloha_real.yaml index 70da3c5610..7c8094da19 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 fd4bf3b595..6ddebab14d 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 840a64e142..d996c3597d 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 3a0975b653..cf5b1f1470 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