@@ -68,9 +68,6 @@ public class CreateTrainJobRequest{
6868 @ Data
6969 @ ToString
7070 public static class FrameworkReplicasDto {
71- /**范围0-1000*/
72- @ KsYunField (name ="Master" )
73- private Integer Master ;
7471 /**当框架为pytorch时候必填,范围0-1000*/
7572 @ KsYunField (name ="Worker" )
7673 private Integer Worker ;
@@ -83,7 +80,17 @@ public static class FrameworkReplicasDto {
8380 /**当框架Framework=tensorflow时候必填,范围0-1000*/
8481 @ KsYunField (name ="PS" )
8582 private Integer PS ;
83+ /**0-1000*/
84+ @ KsYunField (name ="Master" )
85+ private Integer Master ;
8686 }
87+ /**重启策略,可选值:
88+ · Always
89+ · OnFailure
90+ · Never*/
91+ @ KsYunField (name ="RestartPolicy" )
92+ private String RestartPolicy ;
93+
8794 /**环境变量*/
8895 @ KsYunField (name ="Envs" ,type =2 )
8996 private List <EnvsDto > EnvsList ;
@@ -125,8 +132,8 @@ public static class EnvsDto {
125132 private Integer GPUNumber ;
126133
127134 /**范围0-10000*/
128- @ KsYunField (name ="CpuNum " )
129- private Integer CpuNum ;
135+ @ KsYunField (name ="CPUNum " )
136+ private Integer CPUNum ;
130137
131138 /**0-10000*/
132139 @ KsYunField (name ="Memory" )
@@ -165,7 +172,7 @@ public static class StorageConfigsDto {
165172 private Boolean SelfHealing ;
166173
167174 /**任务是否仅运行在cpu节点上,默认false*/
168- @ KsYunField (name ="RunOnCpu " )
169- private Boolean RunOnCpu ;
175+ @ KsYunField (name ="RunOnCPU " )
176+ private Boolean RunOnCPU ;
170177
171178}
0 commit comments