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(
1291
1291
"""
1292
1292
self .checkpoint_config = checkpoint_config or configs .CheckpointConfig ()
1293
1293
return self
1294
-
1294
+
1295
1295
def with_metric_definitions (
1296
1296
self , metric_definitions : List [MetricDefinition ]
1297
1297
) -> "ModelTrainer" : # noqa: D412
1298
1298
"""Set the metric definitions for the training job.
1299
+
1299
1300
Example:
1301
+
1300
1302
.. code:: python
1301
1303
from sagemaker.modules.train import ModelTrainer
1302
1304
from sagemaker.modules.configs import MetricDefinition
@@ -1309,6 +1311,7 @@ def with_metric_definitions(
1309
1311
model_trainer = ModelTrainer(
1310
1312
...
1311
1313
).with_metric_definitions(metric_definitions)
1314
+
1312
1315
Args:
1313
1316
metric_definitions (List[MetricDefinition]):
1314
1317
The metric definitions for the training job.
You can’t perform that action at this time.
0 commit comments