mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-15 08:39:49 +00:00
simple eval
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user