From 90a2a5d5122fc46eafbe03323fa403fa064c1cf1 Mon Sep 17 00:00:00 2001 From: Tianzong Cheng Date: Sat, 20 Sep 2025 14:43:03 +0800 Subject: [PATCH] Fix incorrect teacher obs normalizer input size --- rsl_rl/modules/student_teacher_recurrent.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/rsl_rl/modules/student_teacher_recurrent.py b/rsl_rl/modules/student_teacher_recurrent.py index 964a2dcd..bba4bd34 100644 --- a/rsl_rl/modules/student_teacher_recurrent.py +++ b/rsl_rl/modules/student_teacher_recurrent.py @@ -82,8 +82,7 @@ def __init__( self.memory_t = Memory( num_teacher_obs, type=rnn_type, num_layers=rnn_num_layers, hidden_size=rnn_hidden_dim ) - num_teacher_obs = rnn_hidden_dim - self.teacher = MLP(num_teacher_obs, num_actions, teacher_hidden_dims, activation) + self.teacher = MLP(rnn_hidden_dim, num_actions, teacher_hidden_dims, activation) # teacher observation normalization self.teacher_obs_normalization = teacher_obs_normalization