admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-23 20:50:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

simple eval

Browse Source
This commit is contained in:
Pepijn
2025-08-31 13:43:09 +02:00
parent 7da15ba069
commit d01b060d24
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/rlearn/configuration_rlearn.py
+2
View File
@@ -60,6 +60,8 @@ class RLearNConfig(PreTrainedConfig):
learning_rate: float = 1e-3 learning_rate: float = 1e-3
weight_decay: float = 0.01 weight_decay: float = 0.01
use_tanh_head = False
# Performance optimizations # Performance optimizations
use_amp: bool = True # Mixed precision training for speed boost use_amp: bool = True # Mixed precision training for speed boost
compile_model: bool = True # torch.compile for additional speedup compile_model: bool = True # torch.compile for additional speedup