mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-22 12:09:42 +00:00
simple eval
This commit is contained in:
@@ -59,6 +59,8 @@ class RLearNConfig(PreTrainedConfig):
|
|||||||
# Training
|
# Training
|
||||||
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
|
||||||
|
|||||||
Reference in New Issue
Block a user