admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-22 09:21:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

cleanup

Browse Source
This commit is contained in:
Pepijn
2025-11-25 11:49:27 +01:00
parent 3d28dc3681
commit c2c0dbf52e
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/scripts/lerobot_train.py
-2
View File
@@ -263,7 +263,6 @@ def train(cfg: TrainPipelineConfig, accelerator: Accelerator | None = None):
if is_main_process: if is_main_process:
logging.info("Creating optimizer and scheduler") logging.info("Creating optimizer and scheduler")
optimizer, lr_scheduler = make_optimizer_and_scheduler(cfg, policy) optimizer, lr_scheduler = make_optimizer_and_scheduler(cfg, policy)
grad_scaler = GradScaler(device.type, enabled=cfg.policy.use_amp)
# Load reward model for RA-BC if enabled # Load reward model for RA-BC if enabled
rabc_weight_computer = None rabc_weight_computer = None
@@ -395,7 +394,6 @@ def train(cfg: TrainPipelineConfig, accelerator: Accelerator | None = None):
cfg.optimizer.grad_clip_norm, cfg.optimizer.grad_clip_norm,
accelerator=accelerator, accelerator=accelerator,
lr_scheduler=lr_scheduler, lr_scheduler=lr_scheduler,
use_amp=cfg.policy.use_amp,
rabc_weight_computer=rabc_weight_computer, rabc_weight_computer=rabc_weight_computer,
) )