mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-22 01:11:57 +00:00
simple eval
This commit is contained in:
@@ -59,8 +59,9 @@ 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
|
# Backward compatibility field (not used in current implementation)
|
||||||
|
use_tanh_head: bool = 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