File tree Expand file tree Collapse file tree 1 file changed +2
-0
lines changed
src/sagemaker/modules/train Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Original file line number Diff line number Diff line change @@ -240,6 +240,7 @@ class ModelTrainer(BaseModel):
240240 _infra_check_config : Optional [InfraCheckConfig ] = PrivateAttr (default = None )
241241 _session_chaining_config : Optional [SessionChainingConfig ] = PrivateAttr (default = None )
242242 _remote_debug_config : Optional [RemoteDebugConfig ] = PrivateAttr (default = None )
243+ _metric_definitions : Optional [List [MetricDefinition ]] = PrivateAttr (default = None )
243244
244245 _temp_recipe_train_dir : Optional [TemporaryDirectory ] = PrivateAttr (default = None )
245246
@@ -697,6 +698,7 @@ def train(
697698 training_image_config = self .training_image_config ,
698699 container_entrypoint = container_entrypoint ,
699700 container_arguments = container_arguments ,
701+ metric_definitions = self ._metric_definitions ,
700702 )
701703
702704 resource_config = self .compute ._to_resource_config ()
You can’t perform that action at this time.
0 commit comments