Skip to content

Commit f61c6b0

Browse files
authored
Update model_trainer.py to generate the doc
1 parent 3fd063c commit f61c6b0

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/sagemaker/modules/train/model_trainer.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1300,14 +1300,17 @@ def with_metric_definitions(
13001300
Example:
13011301
13021302
.. code:: python
1303+
13031304
from sagemaker.modules.train import ModelTrainer
13041305
from sagemaker.modules.configs import MetricDefinition
1306+
13051307
metric_definitions = [
13061308
MetricDefinition(
13071309
name="loss",
13081310
regex="Loss: (.*?)",
13091311
)
13101312
]
1313+
13111314
model_trainer = ModelTrainer(
13121315
...
13131316
).with_metric_definitions(metric_definitions)

0 commit comments

Comments
 (0)