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)
def Args(parser): _AddSubmitTrainingArgs(parser) flags.AddCustomContainerFlags(parser) parser.display_info.AddFormat(jobs_util.JOB_FORMAT)