Python JobSet примеры использования

Пример #1

Файл: c_api_util.py Проект: ncnnnnn/oneflow

def GetJobSet():
    job_set, error_str = oneflow_internal.GetSerializedJobSet()
    error = text_format.Parse(error_str, error_util.ErrorProto())
    if error.HasField("error_type"):
        raise JobBuildAndInferError(error)
    return text_format.Parse(job_set, job_set_pb.JobSet())

Пример #2

def GetJobSet():
    job_set = oneflow._oneflow_internal.GetSerializedJobSet()
    return text_format.Parse(job_set, job_set_pb.JobSet())

Пример #3

def GetJobSet():
    job_set = oneflow._oneflow_internal.GetSerializedJobSet()
    ret = job_set_pb.JobSet()
    ret.ParseFromString(job_set)
    return ret

Пример #4

def GetJobSet():
    job_set, error = oneflow_api.GetSerializedJobSet()
    if error.has_error_type():
        raise JobBuildAndInferCfgError(error)
    return text_format.Parse(job_set, job_set_pb.JobSet())