@@ -551,6 +551,11 @@ def __init__(self):
551551 :type PathPrefix: String
552552 :param FrameworkReplicas: 框架副本配置
553553 :type PathPrefix: Object
554+ :param RestartPolicy: 重启策略,可选值:
555+ · Always
556+ · OnFailure
557+ · Never
558+ :type PathPrefix: String
554559 :param Envs: 环境变量
555560 :type PathPrefix: Array
556561 :param SupportTensorboard: 是否开启Tensorboard,默认不开启
@@ -567,7 +572,7 @@ def __init__(self):
567572 :type PathPrefix: String
568573 :param GPUNumber: 范围0-10000
569574 :type PathPrefix: Int
570- :param CpuNum : 范围0-10000
575+ :param CPUNum : 范围0-10000
571576 :type PathPrefix: Int
572577 :param Memory: 0-10000
573578 :type PathPrefix: Int
@@ -580,7 +585,7 @@ def __init__(self):
580585 :type PathPrefix: Int
581586 :param SelfHealing: 是否开启自愈,默认true
582587 :type PathPrefix: Boolean
583- :param RunOnCpu : 任务是否仅运行在cpu节点上,默认false
588+ :param RunOnCPU : 任务是否仅运行在cpu节点上,默认false
584589 :type PathPrefix: Boolean
585590 """
586591 self .TrainJobName = None
@@ -592,6 +597,7 @@ def __init__(self):
592597 self .Framework = None
593598 self .ImageSource = None
594599 self .FrameworkReplicas = None
600+ self .RestartPolicy = None
595601 self .Envs = None
596602 self .SupportTensorboard = None
597603 self .ImageId = None
@@ -600,13 +606,13 @@ def __init__(self):
600606 self .ImageTagId = None
601607 self .GPUType = None
602608 self .GPUNumber = None
603- self .CpuNum = None
609+ self .CPUNum = None
604610 self .Memory = None
605611 self .StorageConfigs = None
606612 self .AccessType = None
607613 self .MaxRuntime = None
608614 self .SelfHealing = None
609- self .RunOnCpu = None
615+ self .RunOnCPU = None
610616
611617 def _deserialize (self , params ):
612618 if params .get ("TrainJobName" ):
@@ -627,6 +633,8 @@ def _deserialize(self, params):
627633 self .ImageSource = params .get ("ImageSource" )
628634 if params .get ("FrameworkReplicas" ):
629635 self .FrameworkReplicas = params .get ("FrameworkReplicas" )
636+ if params .get ("RestartPolicy" ):
637+ self .RestartPolicy = params .get ("RestartPolicy" )
630638 if params .get ("Envs" ):
631639 self .Envs = params .get ("Envs" )
632640 if params .get ("SupportTensorboard" ):
@@ -643,8 +651,8 @@ def _deserialize(self, params):
643651 self .GPUType = params .get ("GPUType" )
644652 if params .get ("GPUNumber" ):
645653 self .GPUNumber = params .get ("GPUNumber" )
646- if params .get ("CpuNum " ):
647- self .CpuNum = params .get ("CpuNum " )
654+ if params .get ("CPUNum " ):
655+ self .CPUNum = params .get ("CPUNum " )
648656 if params .get ("Memory" ):
649657 self .Memory = params .get ("Memory" )
650658 if params .get ("StorageConfigs" ):
@@ -655,8 +663,8 @@ def _deserialize(self, params):
655663 self .MaxRuntime = params .get ("MaxRuntime" )
656664 if params .get ("SelfHealing" ):
657665 self .SelfHealing = params .get ("SelfHealing" )
658- if params .get ("RunOnCpu " ):
659- self .RunOnCpu = params .get ("RunOnCpu " )
666+ if params .get ("RunOnCPU " ):
667+ self .RunOnCPU = params .get ("RunOnCPU " )
660668
661669
662670class DescribeTrainJobEventsRequest (AbstractModel ):
0 commit comments