Python benchmark_context示例

示例#1

文件： imagenet_main.py 项目： jayashreemohan29/models

def main(_):
    with logger.benchmark_context(flags.FLAGS):
        run_imagenet(flags.FLAGS)

示例#2

文件： ncf_estimator_main.py 项目： jiapei100/TensorflowModels

def main(_):
  with logger.benchmark_context(FLAGS), \
       mlperf_helper.LOGGER(FLAGS.output_ml_perf_compliance_logging):
    mlperf_helper.set_ncf_root(os.path.split(os.path.abspath(__file__))[0])
    run_ncf(FLAGS)

示例#3

def main(_):
    with logger.benchmark_context(flags.FLAGS):
        run_census(flags.FLAGS)

示例#4

文件： movielens_main.py 项目： zhou880/Pet-Helper

def main(_):
    with logger.benchmark_context(flags.FLAGS):
        run_movie(flags.FLAGS)

示例#5

文件： deep_speech.py 项目： subzeromot/models

def main(_):
    with logger.benchmark_context(flags_obj):
        run_deep_speech(flags_obj)

示例#6

def main(_):
    with logger.benchmark_context(flags.FLAGS):
        run_transformer(flags.FLAGS)