Python benchmark_contextの例

コード例 #1

ファイル: ncf_keras_main.py プロジェクト: zhangjiulong/models

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])
    if FLAGS.tpu:
      raise ValueError("NCF in Keras does not support TPU for now")
    run_ncf(FLAGS)

コード例 #2

ファイル: logger_test.py プロジェクト: Exscotticus/models

 def test_benchmark_context_failure(self, mock_config_benchmark_logger):
   mock_logger = mock.MagicMock()
   mock_config_benchmark_logger.return_value = mock_logger
   with self.assertRaises(RuntimeError):
     with logger.benchmark_context(None):
       raise RuntimeError("training error")
   mock_logger.on_finish.assert_called_once_with(logger.RUN_STATUS_FAILURE)

コード例 #3

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

コード例 #4

def main(_):
    model_helpers.apply_clean(flags.FLAGS)
    with logger.benchmark_context(flags.FLAGS):
        stats = run(flags.FLAGS)
    logging.info('Run stats:\n%s', stats)

コード例 #5

ファイル: keras_imagenet_main.py プロジェクト: ThomasQin2090/models-1

def main(_):
    model_helpers.apply_clean(flags.FLAGS)
    with logger.benchmark_context(flags.FLAGS):
        return run(flags.FLAGS)

コード例 #6

ファイル: keras_imagenet_main.py プロジェクト: Exscotticus/models

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

コード例 #7

ファイル: transformer_main_triblock.py プロジェクト: domyounglee/Transformer-Summarization

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

コード例 #8

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

コード例 #9

ファイル: wide_deep.py プロジェクト: apple2333cream/models

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

コード例 #10

def main(_):
    with logger.benchmark_context(flags.FLAGS):  # 调用配置Logger
        run_transformer(flags.FLAGS)  # 将参数传递给transformer

コード例 #11

ファイル: keras_imagenet_main.py プロジェクト: rder96/models

def main(_):
  model_helpers.apply_clean(flags.FLAGS)
  with logger.benchmark_context(flags.FLAGS):
    return run(flags.FLAGS)

コード例 #12

def main(_):  #FLAGS
    #pass
    with logger.benchmark_context(flags.FLAGS):
        #pass
        run_retinopathy(flags.FLAGS)

コード例 #13

ファイル: evaluate_model.py プロジェクト: jmsalamy/resnet-test-kungfu

def main(_):
    model_helpers.apply_clean(flags.FLAGS)
    ckpt = "./models/model.ckpt-0060"
    with logger.benchmark_context(flags.FLAGS):
        evaluate_model(flags.FLAGS, ckpt)

コード例 #14

ファイル: deep_speech.py プロジェクト: 812864539/models

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

コード例 #15

ファイル: cifar10_densenet_main.py プロジェクト: Sirius083/mixednet

def main(_):
    with logger.benchmark_context(flags.FLAGS):
        # Note: flags_obj 是在这里定义的
        # print('flags.FLAGS',flags.FLAGS)
        run_cifar(flags.FLAGS)

コード例 #16

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

コード例 #17

ファイル: ncf_main.py プロジェクト: 812864539/models

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)
    mlperf_helper.stitch_ncf()

コード例 #18

def main(_):
    with logger.benchmark_context(FLAGS):
        run_ncf(FLAGS)

コード例 #19

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

コード例 #20

ファイル: movielens_main.py プロジェクト: 812864539/models

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

コード例 #21

ファイル: benchmark_main.py プロジェクト: Fellfalla/tensorflow_models

def main(_):
    with logger.benchmark_context(FLAGS):
        run_keras_model_benchmark(FLAGS)

コード例 #22

ファイル: cifar10_main.py プロジェクト: youzhenfei1995/DistributedTF

def start(_): # Xinyi modified
  with logger.benchmark_context(flags.FLAGS):
    eval_accuracy = run_cifar(flags.FLAGS)
    
    return eval_accuracy, flags.FLAGS.model_id

コード例 #23

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)

コード例 #24

ファイル: logger_test.py プロジェクト: Exscotticus/models

 def test_benchmark_context(self, mock_config_benchmark_logger):
   mock_logger = mock.MagicMock()
   mock_config_benchmark_logger.return_value = mock_logger
   with logger.benchmark_context(None):
     tf.compat.v1.logging.info("start benchmarking")
   mock_logger.on_finish.assert_called_once_with(logger.RUN_STATUS_SUCCESS)

コード例 #25

ファイル: logger_test.py プロジェクト: Bermewda/miniature-vehicle-autonomy-tensorflow-model-on-rpi

 def test_benchmark_context(self, mock_config_benchmark_logger):
   mock_logger = mock.MagicMock()
   mock_config_benchmark_logger.return_value = mock_logger
   with logger.benchmark_context(None):
     logging.info("start benchmarking")
   mock_logger.on_finish.assert_called_once_with(logger.RUN_STATUS_SUCCESS)

コード例 #26

ファイル: retinopathy_main.py プロジェクト: itokeiic/retinopathy_test

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

コード例 #27

ファイル: imagenet_main.py プロジェクト: youlingwangzi/TensorFlow

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

コード例 #28

ファイル: benchmark_main.py プロジェクト: 812864539/models

def main(_):
  with logger.benchmark_context(FLAGS):
    run_keras_model_benchmark(FLAGS)

コード例 #29

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

コード例 #30

ファイル: wide_deep.py プロジェクト: MatsumasaH/TensorflowImageClassifier

def main(_):
    with logger.benchmark_context(flags.FLAGS):
        # Main Function
        run_wide_deep(flags.FLAGS)

コード例 #31

def main(_):
  with logger.benchmark_context(flags.FLAGS):
    stats = run(flags.FLAGS)
  if stats:
    logging.info('Run stats:\n%s', stats)

コード例 #32

ファイル: ncf_main.py プロジェクト: jameslee5656/models

def main(_):
  with logger.benchmark_context(FLAGS):
    run_ncf(FLAGS)

コード例 #33

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

コード例 #34

ファイル: transformer_main.py プロジェクト: Exscotticus/models

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

コード例 #35

ファイル: cifar10_main.py プロジェクト: jameslee5656/models

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

コード例 #36

ファイル: train.py プロジェクト: levanpon98/speech_to_text

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