From 668d493bf997b7d08178d4288a7177d71bb808cf Mon Sep 17 00:00:00 2001 From: Michel Aractingi Date: Wed, 11 Dec 2024 00:22:10 +0100 Subject: [PATCH] Update lerobot/scripts/train_hilserl_classifier.py Co-authored-by: Yoel --- lerobot/scripts/train_hilserl_classifier.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/scripts/train_hilserl_classifier.py b/lerobot/scripts/train_hilserl_classifier.py index 86fa90f2..78659dc8 100644 --- 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))