Remove unnecessary teacher eval call (#139)

Author: ClemensSchwarke

rsl_rl/modules/student_teacher.py

@@ -64,7 +64,6 @@ def __init__(
 
         # Teacher
         self.teacher = MLP(num_teacher_obs, num_actions, teacher_hidden_dims, activation)
-        self.teacher.eval()
         print(f"Teacher MLP: {self.teacher}")
 
         # Teacher observation normalization

rsl_rl/runners/distillation_runner.py

@@ -58,6 +58,7 @@ def __init__(self, env: VecEnv, train_cfg: dict, log_dir: str | None = None, dev
     def learn(self, num_learning_iterations: int, init_at_random_ep_len: bool = False) -> None:
         # Initialize writer
         self._prepare_logging_writer()
+
         # Check if teacher is loaded
         if not self.alg.policy.loaded_teacher:
             raise ValueError("Teacher model parameters not loaded. Please load a teacher model to distill.")