Skip to content

fix #7

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
TingquanGao wants to merge 1 commit into
base: main
Choose a base branch
from
Open

fix #7

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 18 additions & 18 deletions auto_log/autolog.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,20 +27,20 @@
from .device import MemInfo, SubprocessGetMem


class RunConfig:
def __init(self,
class RunConfig(object):
def __init__(self,
run_devices="cpu",
ir_optim=False,
enable_tensorrt=False,
enable_mkldnn=False,
cpu_threads=0,
enable_mem_optim=True):

self.run_devices = run_devices
self.ir_optim = ir_optim
self.enable_mkldnn = enable_mkldnn
self.enable_tensorrt = enable_tensorrt
self.cpu_math_library_num_threads = self.cpu_threads
self.cpu_math_library_num_threads = cpu_threads
self.enable_mem_optim = enable_mem_optim


Expand All @@ -52,14 +52,14 @@ def __init__(self,
data_shape,
save_path,
inference_config=None,
pids=None,
process_name=None,
gpu_ids=None,
pids=None,
process_name=None,
gpu_ids=None,
time_keys=['preprocess_time', 'inference_time', 'postprocess_time'],
warmup=0,
logger=None,
**kwargs):
super(AutoLogger, self).__init__(**kwargs)
super().__init__(**kwargs)
self.autolog_version = 1.0
self.save_path = save_path
self.model_name = model_name
Expand All @@ -74,14 +74,14 @@ def __init__(self,
self.times = Times(keys=time_keys,warmup=warmup)

self.get_paddle_info()

self.logger = self.init_logger() if logger is None else logger

self.get_mem = SubprocessGetMem(pid=pids, gpu_id=gpu_ids)
self.start_subprocess_get_mem()
self.pids = pids
self.gpu_ids = gpu_ids

def start_subprocess_get_mem(self):
self.get_mem.get_mem_subprocess_run(0.2)

Expand All @@ -95,10 +95,10 @@ def end_subprocess_get_mem(self):
else:
self.gpu_infos = gpu_infos[str(self.gpu_ids)]
return self.cpu_infos, self.gpu_infos

def init_logger(self, name='root', log_level=logging.DEBUG):
log_file = self.save_path

logger = logging.getLogger(name)

formatter = logging.Formatter(
Expand Down Expand Up @@ -138,7 +138,7 @@ def parse_config(self, config) -> dict:
)
elif type(config) is dict:
config_status['runtime_device'] = config['runtime_device'] if 'runtime_device' in config else None
config_status['ir_optim'] = config['ir_optim'] if 'ir_optim' in config else None
config_status['ir_optim'] = config['ir_optim'] if 'ir_optim' in config else None
config_status['enable_tensorrt'] = config['enable_tensorrt'] if 'enable_tensorrt' in config else None
config_status['precision'] = config['precision'] if 'precision' in config else None
config_status['enable_mkldnn'] = config['enable_mkldnn'] if 'enable_mkldnn' in config else None
Expand Down Expand Up @@ -178,10 +178,10 @@ def report(self, identifier=None):

# report memory
cpu_infos, gpu_infos = self.end_subprocess_get_mem()

cpu_rss_mb = self.cpu_infos['cpu_rss']
gpu_rss_mb = self.gpu_infos['memory.used'] if self.gpu_ids is not None else None
gpu_util = self.gpu_infos['utilization.gpu'] if self.gpu_ids is not None else None
gpu_rss_mb = self.gpu_infos['memory.used'] if self.gpu_ids is not None else None
gpu_util = self.gpu_infos['utilization.gpu'] if self.gpu_ids is not None else None

# report env
envs = get_env_info()
Expand Down Expand Up @@ -239,7 +239,7 @@ def print_help(self):
"""
print function help
"""
print("""Usage:
print("""Usage:
==== Print inference benchmark logs. ====
config = paddle.inference.Config()
model_info = {'model_name': 'resnet50'
Expand All @@ -264,4 +264,4 @@ def print_help(self):
# print(envs['os_info'])
# get_cudnn_info()
# print(envs['cudnn_version'])