Skip to content

Commit

Permalink
Merge pull request #237 from zouharvi/patch-typo-1
Browse files Browse the repository at this point in the history
Fix model typo in train CLI - RankingMetric
  • Loading branch information
ricardorei authored Dec 29, 2024
2 parents ffe20c6 + 759c020 commit 2af5ff3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion comet/cli/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -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),
Expand Down

0 comments on commit 2af5ff3

Please sign in to comment.