From 040f0bd1010ded4171dfdaa12997f065136d1c15 Mon Sep 17 00:00:00 2001
From: KasparSLT <133706781+KasparSLT@users.noreply.github.com>
Date: Mon, 25 Nov 2024 21:05:13 +0100
Subject: [PATCH] Rename deprecated argument (temporal_ensemble_momentum)
 (#490)

---
 lerobot/configs/policy/act_aloha_real.yaml | 2 +-
 lerobot/configs/policy/act_koch_real.yaml  | 2 +-
 lerobot/configs/policy/act_moss_real.yaml  | 2 +-
 lerobot/configs/policy/act_so100_real.yaml | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/lerobot/configs/policy/act_aloha_real.yaml b/lerobot/configs/policy/act_aloha_real.yaml
index 70da3c561..7c8094da1 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 fd4bf3b59..6ddebab14 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 840a64e14..d996c3597 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 3a0975b65..cf5b1f147 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