ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-03 12:21:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

hls_gaus true

Browse Source
This commit is contained in:
Pepijn
2025-09-01 13:56:08 +02:00
parent f79fdf7205
commit cb0fb8ad15
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/lerobot/policies/rlearn/configuration_rlearn.py
View File

@@ -57,7 +57,7 @@ class RLearNConfig(PreTrainedConfig):
num_register_tokens: int = 4
# --- reward head options ---
use_categorical_rewards: bool = True # classification over bins
use_categorical_rewards: bool = False # classification over bins
num_reward_bins: int = 25
reward_min_value: float = 0.0 # for HL-Gauss range
reward_max_value: float = 1.0
@@ -65,7 +65,7 @@ class RLearNConfig(PreTrainedConfig):
hl_gauss_num_bins: int = 25 # histogram resolution
# Inference-time subsampling and regularization
inference_stride: int = 1 # in forward
inference_stride: int = 1 # inference_stride is an extra, second downsampling applied in forward after window sampling/rewind. Keep it at 1 to disable extra skipping
frame_dropout_p: float = 0.10
# Training