@@ -57,6 +57,15 @@ public final class GoogleCloudMlV1TrainingInput extends com.google.api.client.js
57
57
@ com .google .api .client .util .Key
58
58
private java .lang .String jobDir ;
59
59
60
+ /**
61
+ * Optional. The configuration for master.
62
+ *
63
+ * Only one of `masterConfig.imageUri` and `runtimeVersion` should be set.
64
+ * The value may be {@code null}.
65
+ */
66
+ @ com .google .api .client .util .Key
67
+ private GoogleCloudMlV1ReplicaConfig masterConfig ;
68
+
60
69
/**
61
70
* Optional. Specifies the type of virtual machine to use for your training job's master worker.
62
71
*
@@ -99,6 +108,16 @@ public final class GoogleCloudMlV1TrainingInput extends com.google.api.client.js
99
108
@ com .google .api .client .util .Key
100
109
private java .util .List <java .lang .String > packageUris ;
101
110
111
+ /**
112
+ * Optional. The config of parameter servers.
113
+ *
114
+ * If `parameterServerConfig.imageUri` has not been set, the value of `masterConfig.imageUri` will
115
+ * be used.
116
+ * The value may be {@code null}.
117
+ */
118
+ @ com .google .api .client .util .Key
119
+ private GoogleCloudMlV1ReplicaConfig parameterServerConfig ;
120
+
102
121
/**
103
122
* Optional. The number of parameter server replicas to use for the training job. Each replica in
104
123
* the cluster will be of the type specified in `parameter_server_type`.
@@ -166,6 +185,15 @@ public final class GoogleCloudMlV1TrainingInput extends com.google.api.client.js
166
185
@ com .google .api .client .util .Key
167
186
private java .lang .String scaleTier ;
168
187
188
+ /**
189
+ * Optional. The configrations for workers.
190
+ *
191
+ * If `workerConfig.imageUri` has not been set, the value of `masterConfig.imageUri` will be used.
192
+ * The value may be {@code null}.
193
+ */
194
+ @ com .google .api .client .util .Key
195
+ private GoogleCloudMlV1ReplicaConfig workerConfig ;
196
+
169
197
/**
170
198
* Optional. The number of worker replicas to use for the training job. Each replica in the
171
199
* cluster will be of the type specified in `worker_type`.
@@ -248,6 +276,27 @@ public GoogleCloudMlV1TrainingInput setJobDir(java.lang.String jobDir) {
248
276
return this ;
249
277
}
250
278
279
+ /**
280
+ * Optional. The configuration for master.
281
+ *
282
+ * Only one of `masterConfig.imageUri` and `runtimeVersion` should be set.
283
+ * @return value or {@code null} for none
284
+ */
285
+ public GoogleCloudMlV1ReplicaConfig getMasterConfig () {
286
+ return masterConfig ;
287
+ }
288
+
289
+ /**
290
+ * Optional. The configuration for master.
291
+ *
292
+ * Only one of `masterConfig.imageUri` and `runtimeVersion` should be set.
293
+ * @param masterConfig masterConfig or {@code null} for none
294
+ */
295
+ public GoogleCloudMlV1TrainingInput setMasterConfig (GoogleCloudMlV1ReplicaConfig masterConfig ) {
296
+ this .masterConfig = masterConfig ;
297
+ return this ;
298
+ }
299
+
251
300
/**
252
301
* Optional. Specifies the type of virtual machine to use for your training job's master worker.
253
302
*
@@ -338,6 +387,29 @@ public GoogleCloudMlV1TrainingInput setPackageUris(java.util.List<java.lang.Stri
338
387
return this ;
339
388
}
340
389
390
+ /**
391
+ * Optional. The config of parameter servers.
392
+ *
393
+ * If `parameterServerConfig.imageUri` has not been set, the value of `masterConfig.imageUri` will
394
+ * be used.
395
+ * @return value or {@code null} for none
396
+ */
397
+ public GoogleCloudMlV1ReplicaConfig getParameterServerConfig () {
398
+ return parameterServerConfig ;
399
+ }
400
+
401
+ /**
402
+ * Optional. The config of parameter servers.
403
+ *
404
+ * If `parameterServerConfig.imageUri` has not been set, the value of `masterConfig.imageUri` will
405
+ * be used.
406
+ * @param parameterServerConfig parameterServerConfig or {@code null} for none
407
+ */
408
+ public GoogleCloudMlV1TrainingInput setParameterServerConfig (GoogleCloudMlV1ReplicaConfig parameterServerConfig ) {
409
+ this .parameterServerConfig = parameterServerConfig ;
410
+ return this ;
411
+ }
412
+
341
413
/**
342
414
* Optional. The number of parameter server replicas to use for the training job. Each replica in
343
415
* the cluster will be of the type specified in `parameter_server_type`.
@@ -493,6 +565,27 @@ public GoogleCloudMlV1TrainingInput setScaleTier(java.lang.String scaleTier) {
493
565
return this ;
494
566
}
495
567
568
+ /**
569
+ * Optional. The configrations for workers.
570
+ *
571
+ * If `workerConfig.imageUri` has not been set, the value of `masterConfig.imageUri` will be used.
572
+ * @return value or {@code null} for none
573
+ */
574
+ public GoogleCloudMlV1ReplicaConfig getWorkerConfig () {
575
+ return workerConfig ;
576
+ }
577
+
578
+ /**
579
+ * Optional. The configrations for workers.
580
+ *
581
+ * If `workerConfig.imageUri` has not been set, the value of `masterConfig.imageUri` will be used.
582
+ * @param workerConfig workerConfig or {@code null} for none
583
+ */
584
+ public GoogleCloudMlV1TrainingInput setWorkerConfig (GoogleCloudMlV1ReplicaConfig workerConfig ) {
585
+ this .workerConfig = workerConfig ;
586
+ return this ;
587
+ }
588
+
496
589
/**
497
590
* Optional. The number of worker replicas to use for the training job. Each replica in the
498
591
* cluster will be of the type specified in `worker_type`.
0 commit comments