File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed
src/sagemaker/modules/train Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -1291,12 +1291,14 @@ def with_checkpoint_config(
12911291 """
12921292 self .checkpoint_config = checkpoint_config or configs .CheckpointConfig ()
12931293 return self
1294-
1294+
12951295 def with_metric_definitions (
12961296 self , metric_definitions : List [MetricDefinition ]
12971297 ) -> "ModelTrainer" : # noqa: D412
12981298 """Set the metric definitions for the training job.
1299+
12991300 Example:
1301+
13001302 .. code:: python
13011303 from sagemaker.modules.train import ModelTrainer
13021304 from sagemaker.modules.configs import MetricDefinition
@@ -1309,6 +1311,7 @@ def with_metric_definitions(
13091311 model_trainer = ModelTrainer(
13101312 ...
13111313 ).with_metric_definitions(metric_definitions)
1314+
13121315 Args:
13131316 metric_definitions (List[MetricDefinition]):
13141317 The metric definitions for the training job.
You can’t perform that action at this time.
0 commit comments