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):
240
240
_infra_check_config : Optional [InfraCheckConfig ] = PrivateAttr (default = None )
241
241
_session_chaining_config : Optional [SessionChainingConfig ] = PrivateAttr (default = None )
242
242
_remote_debug_config : Optional [RemoteDebugConfig ] = PrivateAttr (default = None )
243
+ _metric_definitions : Optional [List [MetricDefinition ]] = PrivateAttr (default = None )
243
244
244
245
_temp_recipe_train_dir : Optional [TemporaryDirectory ] = PrivateAttr (default = None )
245
246
@@ -697,6 +698,7 @@ def train(
697
698
training_image_config = self .training_image_config ,
698
699
container_entrypoint = container_entrypoint ,
699
700
container_arguments = container_arguments ,
701
+ metric_definitions = self ._metric_definitions ,
700
702
)
701
703
702
704
resource_config = self .compute ._to_resource_config ()
You can’t perform that action at this time.
0 commit comments