From e9ef46f13441b8941d51ce42e58657173cd9f8e8 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 8dea68c67..ea8336a9f 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))