We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 099535f commit e218d21Copy full SHA for e218d21
huggingface-sagemaker/steps/training/model_trainer.py
@@ -154,6 +154,8 @@ def model_trainer(
154
eval_results = trainer.evaluate(metric_key_prefix="")
155
156
# Log the evaluation results in model control plane
157
- log_artifact_metadata(output_name="model", metrics=eval_results)
+ log_artifact_metadata(
158
+ artifact_name="model", metadata={"metrics": eval_results}
159
+ )
160
161
return model, tokenizer
0 commit comments