AWS SDK for C++

AWS SDK for C++ Version 1.11.719

Loading...
Searching...
No Matches
CreateTrainingJobRequest.h
1
6#pragma once
7#include <aws/core/utils/memory/stl/AWSMap.h>
8#include <aws/core/utils/memory/stl/AWSString.h>
9#include <aws/core/utils/memory/stl/AWSVector.h>
10#include <aws/sagemaker/SageMakerRequest.h>
11#include <aws/sagemaker/SageMaker_EXPORTS.h>
12#include <aws/sagemaker/model/AlgorithmSpecification.h>
13#include <aws/sagemaker/model/Channel.h>
14#include <aws/sagemaker/model/CheckpointConfig.h>
15#include <aws/sagemaker/model/DebugHookConfig.h>
16#include <aws/sagemaker/model/DebugRuleConfiguration.h>
17#include <aws/sagemaker/model/ExperimentConfig.h>
18#include <aws/sagemaker/model/InfraCheckConfig.h>
19#include <aws/sagemaker/model/MlflowConfig.h>
20#include <aws/sagemaker/model/ModelPackageConfig.h>
21#include <aws/sagemaker/model/OutputDataConfig.h>
22#include <aws/sagemaker/model/ProfilerConfig.h>
23#include <aws/sagemaker/model/ProfilerRuleConfiguration.h>
24#include <aws/sagemaker/model/RemoteDebugConfig.h>
25#include <aws/sagemaker/model/ResourceConfig.h>
26#include <aws/sagemaker/model/RetryStrategy.h>
27#include <aws/sagemaker/model/ServerlessJobConfig.h>
28#include <aws/sagemaker/model/SessionChainingConfig.h>
29#include <aws/sagemaker/model/StoppingCondition.h>
30#include <aws/sagemaker/model/Tag.h>
31#include <aws/sagemaker/model/TensorBoardOutputConfig.h>
32#include <aws/sagemaker/model/VpcConfig.h>
33
34#include <utility>
35
36namespace Aws {
37namespace SageMaker {
38namespace Model {
39
43 public:
44 AWS_SAGEMAKER_API CreateTrainingJobRequest() = default;
45
46 // Service request name is the Operation name which will send this request out,
47 // each operation should has unique request name, so that we can get operation's name from this request.
48 // Note: this is not true for response, multiple operations may have the same response name,
49 // so we can not get operation's name from response.
50 inline virtual const char* GetServiceRequestName() const override { return "CreateTrainingJob"; }
51
52 AWS_SAGEMAKER_API Aws::String SerializePayload() const override;
53
55
57
61 inline const Aws::String& GetTrainingJobName() const { return m_trainingJobName; }
62 inline bool TrainingJobNameHasBeenSet() const { return m_trainingJobNameHasBeenSet; }
63 template <typename TrainingJobNameT = Aws::String>
64 void SetTrainingJobName(TrainingJobNameT&& value) {
65 m_trainingJobNameHasBeenSet = true;
66 m_trainingJobName = std::forward<TrainingJobNameT>(value);
67 }
68 template <typename TrainingJobNameT = Aws::String>
69 CreateTrainingJobRequest& WithTrainingJobName(TrainingJobNameT&& value) {
70 SetTrainingJobName(std::forward<TrainingJobNameT>(value));
71 return *this;
72 }
74
76
90 inline const Aws::Map<Aws::String, Aws::String>& GetHyperParameters() const { return m_hyperParameters; }
91 inline bool HyperParametersHasBeenSet() const { return m_hyperParametersHasBeenSet; }
92 template <typename HyperParametersT = Aws::Map<Aws::String, Aws::String>>
93 void SetHyperParameters(HyperParametersT&& value) {
94 m_hyperParametersHasBeenSet = true;
95 m_hyperParameters = std::forward<HyperParametersT>(value);
96 }
97 template <typename HyperParametersT = Aws::Map<Aws::String, Aws::String>>
98 CreateTrainingJobRequest& WithHyperParameters(HyperParametersT&& value) {
99 SetHyperParameters(std::forward<HyperParametersT>(value));
100 return *this;
101 }
102 template <typename HyperParametersKeyT = Aws::String, typename HyperParametersValueT = Aws::String>
103 CreateTrainingJobRequest& AddHyperParameters(HyperParametersKeyT&& key, HyperParametersValueT&& value) {
104 m_hyperParametersHasBeenSet = true;
105 m_hyperParameters.emplace(std::forward<HyperParametersKeyT>(key), std::forward<HyperParametersValueT>(value));
106 return *this;
107 }
109
111
120 inline const AlgorithmSpecification& GetAlgorithmSpecification() const { return m_algorithmSpecification; }
121 inline bool AlgorithmSpecificationHasBeenSet() const { return m_algorithmSpecificationHasBeenSet; }
122 template <typename AlgorithmSpecificationT = AlgorithmSpecification>
123 void SetAlgorithmSpecification(AlgorithmSpecificationT&& value) {
124 m_algorithmSpecificationHasBeenSet = true;
125 m_algorithmSpecification = std::forward<AlgorithmSpecificationT>(value);
126 }
127 template <typename AlgorithmSpecificationT = AlgorithmSpecification>
128 CreateTrainingJobRequest& WithAlgorithmSpecification(AlgorithmSpecificationT&& value) {
129 SetAlgorithmSpecification(std::forward<AlgorithmSpecificationT>(value));
130 return *this;
131 }
133
135
146 inline const Aws::String& GetRoleArn() const { return m_roleArn; }
147 inline bool RoleArnHasBeenSet() const { return m_roleArnHasBeenSet; }
148 template <typename RoleArnT = Aws::String>
149 void SetRoleArn(RoleArnT&& value) {
150 m_roleArnHasBeenSet = true;
151 m_roleArn = std::forward<RoleArnT>(value);
152 }
153 template <typename RoleArnT = Aws::String>
155 SetRoleArn(std::forward<RoleArnT>(value));
156 return *this;
157 }
159
161
176 inline const Aws::Vector<Channel>& GetInputDataConfig() const { return m_inputDataConfig; }
177 inline bool InputDataConfigHasBeenSet() const { return m_inputDataConfigHasBeenSet; }
178 template <typename InputDataConfigT = Aws::Vector<Channel>>
179 void SetInputDataConfig(InputDataConfigT&& value) {
180 m_inputDataConfigHasBeenSet = true;
181 m_inputDataConfig = std::forward<InputDataConfigT>(value);
182 }
183 template <typename InputDataConfigT = Aws::Vector<Channel>>
185 SetInputDataConfig(std::forward<InputDataConfigT>(value));
186 return *this;
187 }
188 template <typename InputDataConfigT = Channel>
189 CreateTrainingJobRequest& AddInputDataConfig(InputDataConfigT&& value) {
190 m_inputDataConfigHasBeenSet = true;
191 m_inputDataConfig.emplace_back(std::forward<InputDataConfigT>(value));
192 return *this;
193 }
195
197
201 inline const OutputDataConfig& GetOutputDataConfig() const { return m_outputDataConfig; }
202 inline bool OutputDataConfigHasBeenSet() const { return m_outputDataConfigHasBeenSet; }
203 template <typename OutputDataConfigT = OutputDataConfig>
204 void SetOutputDataConfig(OutputDataConfigT&& value) {
205 m_outputDataConfigHasBeenSet = true;
206 m_outputDataConfig = std::forward<OutputDataConfigT>(value);
207 }
208 template <typename OutputDataConfigT = OutputDataConfig>
209 CreateTrainingJobRequest& WithOutputDataConfig(OutputDataConfigT&& value) {
210 SetOutputDataConfig(std::forward<OutputDataConfigT>(value));
211 return *this;
212 }
214
216
225 inline const ResourceConfig& GetResourceConfig() const { return m_resourceConfig; }
226 inline bool ResourceConfigHasBeenSet() const { return m_resourceConfigHasBeenSet; }
227 template <typename ResourceConfigT = ResourceConfig>
228 void SetResourceConfig(ResourceConfigT&& value) {
229 m_resourceConfigHasBeenSet = true;
230 m_resourceConfig = std::forward<ResourceConfigT>(value);
231 }
232 template <typename ResourceConfigT = ResourceConfig>
234 SetResourceConfig(std::forward<ResourceConfigT>(value));
235 return *this;
236 }
238
240
249 inline const VpcConfig& GetVpcConfig() const { return m_vpcConfig; }
250 inline bool VpcConfigHasBeenSet() const { return m_vpcConfigHasBeenSet; }
251 template <typename VpcConfigT = VpcConfig>
252 void SetVpcConfig(VpcConfigT&& value) {
253 m_vpcConfigHasBeenSet = true;
254 m_vpcConfig = std::forward<VpcConfigT>(value);
255 }
256 template <typename VpcConfigT = VpcConfig>
258 SetVpcConfig(std::forward<VpcConfigT>(value));
259 return *this;
260 }
262
264
273 inline const StoppingCondition& GetStoppingCondition() const { return m_stoppingCondition; }
274 inline bool StoppingConditionHasBeenSet() const { return m_stoppingConditionHasBeenSet; }
275 template <typename StoppingConditionT = StoppingCondition>
276 void SetStoppingCondition(StoppingConditionT&& value) {
277 m_stoppingConditionHasBeenSet = true;
278 m_stoppingCondition = std::forward<StoppingConditionT>(value);
279 }
280 template <typename StoppingConditionT = StoppingCondition>
281 CreateTrainingJobRequest& WithStoppingCondition(StoppingConditionT&& value) {
282 SetStoppingCondition(std::forward<StoppingConditionT>(value));
283 return *this;
284 }
286
288
300 inline const Aws::Vector<Tag>& GetTags() const { return m_tags; }
301 inline bool TagsHasBeenSet() const { return m_tagsHasBeenSet; }
302 template <typename TagsT = Aws::Vector<Tag>>
303 void SetTags(TagsT&& value) {
304 m_tagsHasBeenSet = true;
305 m_tags = std::forward<TagsT>(value);
306 }
307 template <typename TagsT = Aws::Vector<Tag>>
309 SetTags(std::forward<TagsT>(value));
310 return *this;
311 }
312 template <typename TagsT = Tag>
314 m_tagsHasBeenSet = true;
315 m_tags.emplace_back(std::forward<TagsT>(value));
316 return *this;
317 }
319
321
329 inline bool GetEnableNetworkIsolation() const { return m_enableNetworkIsolation; }
330 inline bool EnableNetworkIsolationHasBeenSet() const { return m_enableNetworkIsolationHasBeenSet; }
331 inline void SetEnableNetworkIsolation(bool value) {
332 m_enableNetworkIsolationHasBeenSet = true;
333 m_enableNetworkIsolation = value;
334 }
337 return *this;
338 }
340
342
352 inline bool GetEnableInterContainerTrafficEncryption() const { return m_enableInterContainerTrafficEncryption; }
353 inline bool EnableInterContainerTrafficEncryptionHasBeenSet() const { return m_enableInterContainerTrafficEncryptionHasBeenSet; }
355 m_enableInterContainerTrafficEncryptionHasBeenSet = true;
356 m_enableInterContainerTrafficEncryption = value;
357 }
360 return *this;
361 }
363
365
376 inline bool GetEnableManagedSpotTraining() const { return m_enableManagedSpotTraining; }
377 inline bool EnableManagedSpotTrainingHasBeenSet() const { return m_enableManagedSpotTrainingHasBeenSet; }
378 inline void SetEnableManagedSpotTraining(bool value) {
379 m_enableManagedSpotTrainingHasBeenSet = true;
380 m_enableManagedSpotTraining = value;
381 }
384 return *this;
385 }
387
389
393 inline const CheckpointConfig& GetCheckpointConfig() const { return m_checkpointConfig; }
394 inline bool CheckpointConfigHasBeenSet() const { return m_checkpointConfigHasBeenSet; }
395 template <typename CheckpointConfigT = CheckpointConfig>
396 void SetCheckpointConfig(CheckpointConfigT&& value) {
397 m_checkpointConfigHasBeenSet = true;
398 m_checkpointConfig = std::forward<CheckpointConfigT>(value);
399 }
400 template <typename CheckpointConfigT = CheckpointConfig>
401 CreateTrainingJobRequest& WithCheckpointConfig(CheckpointConfigT&& value) {
402 SetCheckpointConfig(std::forward<CheckpointConfigT>(value));
403 return *this;
404 }
406
408
409 inline const DebugHookConfig& GetDebugHookConfig() const { return m_debugHookConfig; }
410 inline bool DebugHookConfigHasBeenSet() const { return m_debugHookConfigHasBeenSet; }
411 template <typename DebugHookConfigT = DebugHookConfig>
412 void SetDebugHookConfig(DebugHookConfigT&& value) {
413 m_debugHookConfigHasBeenSet = true;
414 m_debugHookConfig = std::forward<DebugHookConfigT>(value);
415 }
416 template <typename DebugHookConfigT = DebugHookConfig>
418 SetDebugHookConfig(std::forward<DebugHookConfigT>(value));
419 return *this;
420 }
422
424
428 inline const Aws::Vector<DebugRuleConfiguration>& GetDebugRuleConfigurations() const { return m_debugRuleConfigurations; }
429 inline bool DebugRuleConfigurationsHasBeenSet() const { return m_debugRuleConfigurationsHasBeenSet; }
430 template <typename DebugRuleConfigurationsT = Aws::Vector<DebugRuleConfiguration>>
431 void SetDebugRuleConfigurations(DebugRuleConfigurationsT&& value) {
432 m_debugRuleConfigurationsHasBeenSet = true;
433 m_debugRuleConfigurations = std::forward<DebugRuleConfigurationsT>(value);
434 }
435 template <typename DebugRuleConfigurationsT = Aws::Vector<DebugRuleConfiguration>>
436 CreateTrainingJobRequest& WithDebugRuleConfigurations(DebugRuleConfigurationsT&& value) {
437 SetDebugRuleConfigurations(std::forward<DebugRuleConfigurationsT>(value));
438 return *this;
439 }
440 template <typename DebugRuleConfigurationsT = DebugRuleConfiguration>
441 CreateTrainingJobRequest& AddDebugRuleConfigurations(DebugRuleConfigurationsT&& value) {
442 m_debugRuleConfigurationsHasBeenSet = true;
443 m_debugRuleConfigurations.emplace_back(std::forward<DebugRuleConfigurationsT>(value));
444 return *this;
445 }
447
449
450 inline const TensorBoardOutputConfig& GetTensorBoardOutputConfig() const { return m_tensorBoardOutputConfig; }
451 inline bool TensorBoardOutputConfigHasBeenSet() const { return m_tensorBoardOutputConfigHasBeenSet; }
452 template <typename TensorBoardOutputConfigT = TensorBoardOutputConfig>
453 void SetTensorBoardOutputConfig(TensorBoardOutputConfigT&& value) {
454 m_tensorBoardOutputConfigHasBeenSet = true;
455 m_tensorBoardOutputConfig = std::forward<TensorBoardOutputConfigT>(value);
456 }
457 template <typename TensorBoardOutputConfigT = TensorBoardOutputConfig>
458 CreateTrainingJobRequest& WithTensorBoardOutputConfig(TensorBoardOutputConfigT&& value) {
459 SetTensorBoardOutputConfig(std::forward<TensorBoardOutputConfigT>(value));
460 return *this;
461 }
463
465
466 inline const ExperimentConfig& GetExperimentConfig() const { return m_experimentConfig; }
467 inline bool ExperimentConfigHasBeenSet() const { return m_experimentConfigHasBeenSet; }
468 template <typename ExperimentConfigT = ExperimentConfig>
469 void SetExperimentConfig(ExperimentConfigT&& value) {
470 m_experimentConfigHasBeenSet = true;
471 m_experimentConfig = std::forward<ExperimentConfigT>(value);
472 }
473 template <typename ExperimentConfigT = ExperimentConfig>
474 CreateTrainingJobRequest& WithExperimentConfig(ExperimentConfigT&& value) {
475 SetExperimentConfig(std::forward<ExperimentConfigT>(value));
476 return *this;
477 }
479
481
482 inline const ProfilerConfig& GetProfilerConfig() const { return m_profilerConfig; }
483 inline bool ProfilerConfigHasBeenSet() const { return m_profilerConfigHasBeenSet; }
484 template <typename ProfilerConfigT = ProfilerConfig>
485 void SetProfilerConfig(ProfilerConfigT&& value) {
486 m_profilerConfigHasBeenSet = true;
487 m_profilerConfig = std::forward<ProfilerConfigT>(value);
488 }
489 template <typename ProfilerConfigT = ProfilerConfig>
491 SetProfilerConfig(std::forward<ProfilerConfigT>(value));
492 return *this;
493 }
495
497
501 inline const Aws::Vector<ProfilerRuleConfiguration>& GetProfilerRuleConfigurations() const { return m_profilerRuleConfigurations; }
502 inline bool ProfilerRuleConfigurationsHasBeenSet() const { return m_profilerRuleConfigurationsHasBeenSet; }
503 template <typename ProfilerRuleConfigurationsT = Aws::Vector<ProfilerRuleConfiguration>>
504 void SetProfilerRuleConfigurations(ProfilerRuleConfigurationsT&& value) {
505 m_profilerRuleConfigurationsHasBeenSet = true;
506 m_profilerRuleConfigurations = std::forward<ProfilerRuleConfigurationsT>(value);
507 }
508 template <typename ProfilerRuleConfigurationsT = Aws::Vector<ProfilerRuleConfiguration>>
509 CreateTrainingJobRequest& WithProfilerRuleConfigurations(ProfilerRuleConfigurationsT&& value) {
510 SetProfilerRuleConfigurations(std::forward<ProfilerRuleConfigurationsT>(value));
511 return *this;
512 }
513 template <typename ProfilerRuleConfigurationsT = ProfilerRuleConfiguration>
514 CreateTrainingJobRequest& AddProfilerRuleConfigurations(ProfilerRuleConfigurationsT&& value) {
515 m_profilerRuleConfigurationsHasBeenSet = true;
516 m_profilerRuleConfigurations.emplace_back(std::forward<ProfilerRuleConfigurationsT>(value));
517 return *this;
518 }
520
522
531 inline const Aws::Map<Aws::String, Aws::String>& GetEnvironment() const { return m_environment; }
532 inline bool EnvironmentHasBeenSet() const { return m_environmentHasBeenSet; }
533 template <typename EnvironmentT = Aws::Map<Aws::String, Aws::String>>
534 void SetEnvironment(EnvironmentT&& value) {
535 m_environmentHasBeenSet = true;
536 m_environment = std::forward<EnvironmentT>(value);
537 }
538 template <typename EnvironmentT = Aws::Map<Aws::String, Aws::String>>
540 SetEnvironment(std::forward<EnvironmentT>(value));
541 return *this;
542 }
543 template <typename EnvironmentKeyT = Aws::String, typename EnvironmentValueT = Aws::String>
544 CreateTrainingJobRequest& AddEnvironment(EnvironmentKeyT&& key, EnvironmentValueT&& value) {
545 m_environmentHasBeenSet = true;
546 m_environment.emplace(std::forward<EnvironmentKeyT>(key), std::forward<EnvironmentValueT>(value));
547 return *this;
548 }
550
552
556 inline const RetryStrategy& GetRetryStrategy() const { return m_retryStrategy; }
557 inline bool RetryStrategyHasBeenSet() const { return m_retryStrategyHasBeenSet; }
558 template <typename RetryStrategyT = RetryStrategy>
559 void SetRetryStrategy(RetryStrategyT&& value) {
560 m_retryStrategyHasBeenSet = true;
561 m_retryStrategy = std::forward<RetryStrategyT>(value);
562 }
563 template <typename RetryStrategyT = RetryStrategy>
565 SetRetryStrategy(std::forward<RetryStrategyT>(value));
566 return *this;
567 }
569
571
578 inline const RemoteDebugConfig& GetRemoteDebugConfig() const { return m_remoteDebugConfig; }
579 inline bool RemoteDebugConfigHasBeenSet() const { return m_remoteDebugConfigHasBeenSet; }
580 template <typename RemoteDebugConfigT = RemoteDebugConfig>
581 void SetRemoteDebugConfig(RemoteDebugConfigT&& value) {
582 m_remoteDebugConfigHasBeenSet = true;
583 m_remoteDebugConfig = std::forward<RemoteDebugConfigT>(value);
584 }
585 template <typename RemoteDebugConfigT = RemoteDebugConfig>
586 CreateTrainingJobRequest& WithRemoteDebugConfig(RemoteDebugConfigT&& value) {
587 SetRemoteDebugConfig(std::forward<RemoteDebugConfigT>(value));
588 return *this;
589 }
591
593
597 inline const InfraCheckConfig& GetInfraCheckConfig() const { return m_infraCheckConfig; }
598 inline bool InfraCheckConfigHasBeenSet() const { return m_infraCheckConfigHasBeenSet; }
599 template <typename InfraCheckConfigT = InfraCheckConfig>
600 void SetInfraCheckConfig(InfraCheckConfigT&& value) {
601 m_infraCheckConfigHasBeenSet = true;
602 m_infraCheckConfig = std::forward<InfraCheckConfigT>(value);
603 }
604 template <typename InfraCheckConfigT = InfraCheckConfig>
605 CreateTrainingJobRequest& WithInfraCheckConfig(InfraCheckConfigT&& value) {
606 SetInfraCheckConfig(std::forward<InfraCheckConfigT>(value));
607 return *this;
608 }
610
612
616 inline const SessionChainingConfig& GetSessionChainingConfig() const { return m_sessionChainingConfig; }
617 inline bool SessionChainingConfigHasBeenSet() const { return m_sessionChainingConfigHasBeenSet; }
618 template <typename SessionChainingConfigT = SessionChainingConfig>
619 void SetSessionChainingConfig(SessionChainingConfigT&& value) {
620 m_sessionChainingConfigHasBeenSet = true;
621 m_sessionChainingConfig = std::forward<SessionChainingConfigT>(value);
622 }
623 template <typename SessionChainingConfigT = SessionChainingConfig>
624 CreateTrainingJobRequest& WithSessionChainingConfig(SessionChainingConfigT&& value) {
625 SetSessionChainingConfig(std::forward<SessionChainingConfigT>(value));
626 return *this;
627 }
629
631
634 inline const ServerlessJobConfig& GetServerlessJobConfig() const { return m_serverlessJobConfig; }
635 inline bool ServerlessJobConfigHasBeenSet() const { return m_serverlessJobConfigHasBeenSet; }
636 template <typename ServerlessJobConfigT = ServerlessJobConfig>
637 void SetServerlessJobConfig(ServerlessJobConfigT&& value) {
638 m_serverlessJobConfigHasBeenSet = true;
639 m_serverlessJobConfig = std::forward<ServerlessJobConfigT>(value);
640 }
641 template <typename ServerlessJobConfigT = ServerlessJobConfig>
642 CreateTrainingJobRequest& WithServerlessJobConfig(ServerlessJobConfigT&& value) {
643 SetServerlessJobConfig(std::forward<ServerlessJobConfigT>(value));
644 return *this;
645 }
647
649
652 inline const MlflowConfig& GetMlflowConfig() const { return m_mlflowConfig; }
653 inline bool MlflowConfigHasBeenSet() const { return m_mlflowConfigHasBeenSet; }
654 template <typename MlflowConfigT = MlflowConfig>
655 void SetMlflowConfig(MlflowConfigT&& value) {
656 m_mlflowConfigHasBeenSet = true;
657 m_mlflowConfig = std::forward<MlflowConfigT>(value);
658 }
659 template <typename MlflowConfigT = MlflowConfig>
661 SetMlflowConfig(std::forward<MlflowConfigT>(value));
662 return *this;
663 }
665
667
670 inline const ModelPackageConfig& GetModelPackageConfig() const { return m_modelPackageConfig; }
671 inline bool ModelPackageConfigHasBeenSet() const { return m_modelPackageConfigHasBeenSet; }
672 template <typename ModelPackageConfigT = ModelPackageConfig>
673 void SetModelPackageConfig(ModelPackageConfigT&& value) {
674 m_modelPackageConfigHasBeenSet = true;
675 m_modelPackageConfig = std::forward<ModelPackageConfigT>(value);
676 }
677 template <typename ModelPackageConfigT = ModelPackageConfig>
678 CreateTrainingJobRequest& WithModelPackageConfig(ModelPackageConfigT&& value) {
679 SetModelPackageConfig(std::forward<ModelPackageConfigT>(value));
680 return *this;
681 }
683 private:
684 Aws::String m_trainingJobName;
685
686 Aws::Map<Aws::String, Aws::String> m_hyperParameters;
687
688 AlgorithmSpecification m_algorithmSpecification;
689
690 Aws::String m_roleArn;
691
692 Aws::Vector<Channel> m_inputDataConfig;
693
694 OutputDataConfig m_outputDataConfig;
695
696 ResourceConfig m_resourceConfig;
697
698 VpcConfig m_vpcConfig;
699
700 StoppingCondition m_stoppingCondition;
701
702 Aws::Vector<Tag> m_tags;
703
704 bool m_enableNetworkIsolation{false};
705
706 bool m_enableInterContainerTrafficEncryption{false};
707
708 bool m_enableManagedSpotTraining{false};
709
710 CheckpointConfig m_checkpointConfig;
711
712 DebugHookConfig m_debugHookConfig;
713
714 Aws::Vector<DebugRuleConfiguration> m_debugRuleConfigurations;
715
716 TensorBoardOutputConfig m_tensorBoardOutputConfig;
717
718 ExperimentConfig m_experimentConfig;
719
720 ProfilerConfig m_profilerConfig;
721
722 Aws::Vector<ProfilerRuleConfiguration> m_profilerRuleConfigurations;
723
725
726 RetryStrategy m_retryStrategy;
727
728 RemoteDebugConfig m_remoteDebugConfig;
729
730 InfraCheckConfig m_infraCheckConfig;
731
732 SessionChainingConfig m_sessionChainingConfig;
733
734 ServerlessJobConfig m_serverlessJobConfig;
735
736 MlflowConfig m_mlflowConfig;
737
738 ModelPackageConfig m_modelPackageConfig;
739 bool m_trainingJobNameHasBeenSet = false;
740 bool m_hyperParametersHasBeenSet = false;
741 bool m_algorithmSpecificationHasBeenSet = false;
742 bool m_roleArnHasBeenSet = false;
743 bool m_inputDataConfigHasBeenSet = false;
744 bool m_outputDataConfigHasBeenSet = false;
745 bool m_resourceConfigHasBeenSet = false;
746 bool m_vpcConfigHasBeenSet = false;
747 bool m_stoppingConditionHasBeenSet = false;
748 bool m_tagsHasBeenSet = false;
749 bool m_enableNetworkIsolationHasBeenSet = false;
750 bool m_enableInterContainerTrafficEncryptionHasBeenSet = false;
751 bool m_enableManagedSpotTrainingHasBeenSet = false;
752 bool m_checkpointConfigHasBeenSet = false;
753 bool m_debugHookConfigHasBeenSet = false;
754 bool m_debugRuleConfigurationsHasBeenSet = false;
755 bool m_tensorBoardOutputConfigHasBeenSet = false;
756 bool m_experimentConfigHasBeenSet = false;
757 bool m_profilerConfigHasBeenSet = false;
758 bool m_profilerRuleConfigurationsHasBeenSet = false;
759 bool m_environmentHasBeenSet = false;
760 bool m_retryStrategyHasBeenSet = false;
761 bool m_remoteDebugConfigHasBeenSet = false;
762 bool m_infraCheckConfigHasBeenSet = false;
763 bool m_sessionChainingConfigHasBeenSet = false;
764 bool m_serverlessJobConfigHasBeenSet = false;
765 bool m_mlflowConfigHasBeenSet = false;
766 bool m_modelPackageConfigHasBeenSet = false;
767};
768
769} // namespace Model
770} // namespace SageMaker
771} // namespace Aws
virtual const char * GetServiceRequestName() const override
CreateTrainingJobRequest & WithEnableInterContainerTrafficEncryption(bool value)
CreateTrainingJobRequest & WithResourceConfig(ResourceConfigT &&value)
const Aws::Vector< DebugRuleConfiguration > & GetDebugRuleConfigurations() const
CreateTrainingJobRequest & WithSessionChainingConfig(SessionChainingConfigT &&value)
AWS_SAGEMAKER_API Aws::Http::HeaderValueCollection GetRequestSpecificHeaders() const override
const ModelPackageConfig & GetModelPackageConfig() const
CreateTrainingJobRequest & WithVpcConfig(VpcConfigT &&value)
const SessionChainingConfig & GetSessionChainingConfig() const
CreateTrainingJobRequest & WithEnableNetworkIsolation(bool value)
const Aws::Map< Aws::String, Aws::String > & GetHyperParameters() const
CreateTrainingJobRequest & WithCheckpointConfig(CheckpointConfigT &&value)
CreateTrainingJobRequest & WithModelPackageConfig(ModelPackageConfigT &&value)
CreateTrainingJobRequest & WithDebugRuleConfigurations(DebugRuleConfigurationsT &&value)
AWS_SAGEMAKER_API CreateTrainingJobRequest()=default
const Aws::Vector< Channel > & GetInputDataConfig() const
void SetServerlessJobConfig(ServerlessJobConfigT &&value)
CreateTrainingJobRequest & WithRemoteDebugConfig(RemoteDebugConfigT &&value)
CreateTrainingJobRequest & WithTrainingJobName(TrainingJobNameT &&value)
CreateTrainingJobRequest & WithServerlessJobConfig(ServerlessJobConfigT &&value)
CreateTrainingJobRequest & AddProfilerRuleConfigurations(ProfilerRuleConfigurationsT &&value)
CreateTrainingJobRequest & WithTags(TagsT &&value)
CreateTrainingJobRequest & WithEnableManagedSpotTraining(bool value)
CreateTrainingJobRequest & AddEnvironment(EnvironmentKeyT &&key, EnvironmentValueT &&value)
const AlgorithmSpecification & GetAlgorithmSpecification() const
CreateTrainingJobRequest & AddInputDataConfig(InputDataConfigT &&value)
const TensorBoardOutputConfig & GetTensorBoardOutputConfig() const
CreateTrainingJobRequest & WithExperimentConfig(ExperimentConfigT &&value)
CreateTrainingJobRequest & WithStoppingCondition(StoppingConditionT &&value)
CreateTrainingJobRequest & AddHyperParameters(HyperParametersKeyT &&key, HyperParametersValueT &&value)
const Aws::Map< Aws::String, Aws::String > & GetEnvironment() const
CreateTrainingJobRequest & WithRetryStrategy(RetryStrategyT &&value)
CreateTrainingJobRequest & WithTensorBoardOutputConfig(TensorBoardOutputConfigT &&value)
CreateTrainingJobRequest & WithAlgorithmSpecification(AlgorithmSpecificationT &&value)
void SetDebugRuleConfigurations(DebugRuleConfigurationsT &&value)
CreateTrainingJobRequest & AddTags(TagsT &&value)
const ServerlessJobConfig & GetServerlessJobConfig() const
CreateTrainingJobRequest & AddDebugRuleConfigurations(DebugRuleConfigurationsT &&value)
void SetProfilerRuleConfigurations(ProfilerRuleConfigurationsT &&value)
CreateTrainingJobRequest & WithHyperParameters(HyperParametersT &&value)
void SetSessionChainingConfig(SessionChainingConfigT &&value)
CreateTrainingJobRequest & WithDebugHookConfig(DebugHookConfigT &&value)
void SetTensorBoardOutputConfig(TensorBoardOutputConfigT &&value)
CreateTrainingJobRequest & WithMlflowConfig(MlflowConfigT &&value)
CreateTrainingJobRequest & WithInfraCheckConfig(InfraCheckConfigT &&value)
void SetAlgorithmSpecification(AlgorithmSpecificationT &&value)
const Aws::Vector< ProfilerRuleConfiguration > & GetProfilerRuleConfigurations() const
CreateTrainingJobRequest & WithProfilerRuleConfigurations(ProfilerRuleConfigurationsT &&value)
CreateTrainingJobRequest & WithInputDataConfig(InputDataConfigT &&value)
CreateTrainingJobRequest & WithEnvironment(EnvironmentT &&value)
AWS_SAGEMAKER_API Aws::String SerializePayload() const override
CreateTrainingJobRequest & WithProfilerConfig(ProfilerConfigT &&value)
CreateTrainingJobRequest & WithOutputDataConfig(OutputDataConfigT &&value)
CreateTrainingJobRequest & WithRoleArn(RoleArnT &&value)
Aws::Map< Aws::String, Aws::String > HeaderValueCollection
std::map< K, V, std::less< K >, Aws::Allocator< std::pair< const K, V > > > Map
std::basic_string< char, std::char_traits< char >, Aws::Allocator< char > > String
std::vector< T, Aws::Allocator< T > > Vector