From c2c0dbf52efb7f2cccb0e1ea1f95c59bace6493c Mon Sep 17 00:00:00 2001 From: Pepijn Date: Tue, 25 Nov 2025 11:49:27 +0100 Subject: [PATCH] cleanup --- src/lerobot/scripts/lerobot_train.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/lerobot/scripts/lerobot_train.py b/src/lerobot/scripts/lerobot_train.py index 11ed28bfd..9768bbde6 100644 --- a/src/lerobot/scripts/lerobot_train.py +++ b/src/lerobot/scripts/lerobot_train.py @@ -263,7 +263,6 @@ def train(cfg: TrainPipelineConfig, accelerator: Accelerator | None = None): if is_main_process: logging.info("Creating optimizer and scheduler") 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 rabc_weight_computer = None @@ -395,7 +394,6 @@ def train(cfg: TrainPipelineConfig, accelerator: Accelerator | None = None): cfg.optimizer.grad_clip_norm, accelerator=accelerator, lr_scheduler=lr_scheduler, - use_amp=cfg.policy.use_amp, rabc_weight_computer=rabc_weight_computer, )