def Args(cls, parser):
   _AddSubmitTrainingArgs(parser)
   flags.AddKmsKeyFlag(parser, 'job')
   flags.NETWORK.AddToParser(parser)
   flags.AddCustomContainerFlags(
       parser, support_tpu_tf_version=cls._SUPPORT_TPU_TF_VERSION)
   parser.display_info.AddFormat(jobs_util.JOB_FORMAT)
Beispiel #2
0
 def Args(parser):
   _AddSubmitTrainingArgs(parser)
   flags.AddCustomContainerFlags(parser)
   parser.display_info.AddFormat(jobs_util.JOB_FORMAT)