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 3fd063c commit f61c6b0Copy full SHA for f61c6b0
src/sagemaker/modules/train/model_trainer.py
@@ -1300,14 +1300,17 @@ def with_metric_definitions(
1300
Example:
1301
1302
.. code:: python
1303
+
1304
from sagemaker.modules.train import ModelTrainer
1305
from sagemaker.modules.configs import MetricDefinition
1306
1307
metric_definitions = [
1308
MetricDefinition(
1309
name="loss",
1310
regex="Loss: (.*?)",
1311
)
1312
]
1313
1314
model_trainer = ModelTrainer(
1315
...
1316
).with_metric_definitions(metric_definitions)
0 commit comments