Skip to content

Commit e218d21

Browse files
committed
fix artifact logging function
1 parent 099535f commit e218d21

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

huggingface-sagemaker/steps/training/model_trainer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,6 +154,8 @@ def model_trainer(
154154
eval_results = trainer.evaluate(metric_key_prefix="")
155155

156156
# Log the evaluation results in model control plane
157-
log_artifact_metadata(output_name="model", metrics=eval_results)
157+
log_artifact_metadata(
158+
artifact_name="model", metadata={"metrics": eval_results}
159+
)
158160

159161
return model, tokenizer

0 commit comments

Comments
 (0)