From d01b060d240f08c6aa027b8c7363321d8feebdd6 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Sun, 31 Aug 2025 13:43:09 +0200 Subject: [PATCH] simple eval --- src/lerobot/policies/rlearn/configuration_rlearn.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/lerobot/policies/rlearn/configuration_rlearn.py b/src/lerobot/policies/rlearn/configuration_rlearn.py index 22609f06f..89d12cb55 100644 --- a/src/lerobot/policies/rlearn/configuration_rlearn.py +++ b/src/lerobot/policies/rlearn/configuration_rlearn.py @@ -59,6 +59,8 @@ class RLearNConfig(PreTrainedConfig): # Training learning_rate: float = 1e-3 weight_decay: float = 0.01 + + use_tanh_head = False # Performance optimizations use_amp: bool = True # Mixed precision training for speed boost