From 759c0203440f67b6ca8504b191ee418a32c7723f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Vil=C3=A9m=20Zouhar?= Date: Fri, 13 Dec 2024 20:56:55 +0100 Subject: [PATCH] fix model typo in train CLI --- comet/cli/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comet/cli/train.py b/comet/cli/train.py index fc645bc..6ccc519 100644 --- a/comet/cli/train.py +++ b/comet/cli/train.py @@ -142,7 +142,7 @@ def initialize_model(configs): ) if configs.load_from_checkpoint is not None: logger.info(f"Loading weights from {configs.load_from_checkpoint}.") - model = ReferencelessRegression.load_from_checkpoint( + model = RankingMetric.load_from_checkpoint( checkpoint_path=configs.load_from_checkpoint, strict=configs.strict_load, **namespace_to_dict(configs.ranking_metric.init_args),