From 599218fe9a82b4e89f421aeb2e1fd5951358ba1f Mon Sep 17 00:00:00 2001 From: Pepijn Date: Sat, 30 Aug 2025 14:41:15 +0200 Subject: [PATCH] use rewind --- src/lerobot/policies/rlearn/configuration_rlearn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/policies/rlearn/configuration_rlearn.py b/src/lerobot/policies/rlearn/configuration_rlearn.py index 74a227b11..9a1d41f17 100644 --- a/src/lerobot/policies/rlearn/configuration_rlearn.py +++ b/src/lerobot/policies/rlearn/configuration_rlearn.py @@ -69,7 +69,7 @@ class RLearNConfig(PreTrainedConfig): compile_model: bool = True # torch.compile for additional speedup # ReWiND-specific parameters - use_video_rewind: bool = False # Enable video rewinding augmentation + use_video_rewind: bool = True # Enable video rewinding augmentation rewind_prob: float = 0.8 # Probability of applying rewind to each sample (paper: ~80%) rewind_last3_prob: float = 0.1 # Of the rewinds, 10% only rewind the last 3 frames use_mismatch_loss: bool = False # Enable mismatched language-video loss