Update lerobot/scripts/train_hilserl_classifier.py

Co-authored-by: Yoel <yoel.chornton@gmail.com>
huggingface · Dec 10, 2024 · e9ef46f · e9ef46f
1 parent d26a9b1
commit e9ef46f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lerobot/scripts/train_hilserl_classifier.py b/lerobot/scripts/train_hilserl_classifier.py
@@ -170,7 +170,7 @@ def validate(model, val_loader, criterion, device, logger, cfg, num_samples_to_l
     return accuracy, eval_info
 
 
-@hydra.main(version_base="1.2", config_path="../configs", config_name="classifier")
+@hydra.main(version_base="1.2", config_path="../configs", config_name="hilserl_classifier")
 def train(cfg: DictConfig) -> None:
     # Main training pipeline with support for resuming training
     logging.info(OmegaConf.to_yaml(cfg))