admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 02:06:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

use rewind

Browse Source
This commit is contained in:
Pepijn
2025-08-30 14:41:15 +02:00
parent 2507341a32
commit 599218fe9a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/rlearn/configuration_rlearn.py
+1 -1
View File
@@ -69,7 +69,7 @@ class RLearNConfig(PreTrainedConfig):
compile_model: bool = True # torch.compile for additional speedup compile_model: bool = True # torch.compile for additional speedup
# ReWiND-specific parameters # 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_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 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 use_mismatch_loss: bool = False # Enable mismatched language-video loss