diff --git a/python/examples/bert/parse_profile.py b/python/examples/bert/parse_profile.py index fcc6e05b5ecb0120ae6f31c1f6b8ba8549a5d92d..5ab5d84e4e7031c6b7a5a1381834b852546347ab 100644 --- a/python/examples/bert/parse_profile.py +++ b/python/examples/bert/parse_profile.py @@ -1,34 +1,34 @@ import sys import os ## general info -cuda_version = "" -cudnn_version = "" -trt_version = "" -python_version = "" -gcc_version = "" -paddle_version = "" -cpu = "" -gpu = "" +cuda_version = "10.1" +cudnn_version = "7" +trt_version = "6" +python_version = "3.6" +gcc_version = "8.2" +paddle_version = "2.0.0" +cpu = "Xeon 6148" +gpu = "P4" xpu = "" api = "" owner = "" ## model info -model_name = "" -model_type = "" -model_source = "" +model_name = "bert_seq128" +model_type = "static" +model_source = "paddlehub" model_url = "" ## data info -batch_size = "" -num_of_samples = "" -input_shape = "" +batch_size = "1" +num_of_samples = "1" +input_shape = "none" ## conf info -runtime_device = "" -ir_optim = "" -enable_memory_optim = "" -enable_tensorrt = "" -precision = "" -enable_mkldnn = "" -cpu_math_library_num_threads = "" +runtime_device = "cpu" +ir_optim = "true" +enable_memory_optim = "true" +enable_tensorrt = "false" +precision = "fp32" +enable_mkldnn = "false" +cpu_math_library_num_threads = "1" ## acc info acc1 = "" acc5 = ""