def app(argv):
telemetry = start_session_telemetry()
parser = get_common_argument_parser()
args = parser.parse_args(args=argv)
check_dependencies(args)
if not args.config:
_update_config_path(args)
config = Config.read_config(args.config)
config.configure_params(args.ac_config)
config.update_from_args(args)
if config.engine.type == 'simplified' and args.evaluate:
raise Exception('Can not make evaluation in simplified mode')
log_dir = _create_log_path(config)
init_logger(level=args.log_level,
file_name=os.path.join(log_dir, 'log.txt'),
progress_bar=args.pbar)
logger.info('Output log dir: {}'.format(log_dir))
metrics = optimize(config)
if metrics and logger.progress_bar_disabled:
for name, value in metrics.items():
logger.info('{: <27s}: {}'.format(name, value))
end_session_telemetry(telemetry)
def test_load_tool_config(config_name, argv):
parser = get_common_argument_parser()
argv = argv.split()
argv[-1] = ENGINE_CONFIG_PATH.joinpath('mobilenet-ssd.json').as_posix()
args = parser.parse_args(args=argv)
tool_config_path = TOOL_CONFIG_PATH.joinpath(config_name).as_posix()
config = Config.read_config(tool_config_path)
config.configure_params()
config.update_from_args(args)
assert config.model.model == argv[5]
assert config.model.weights == argv[3]
def app(argv):
telemetry = start_session_telemetry()
parser = get_common_argument_parser()
args = parser.parse_args(args=argv)
check_dependencies(args)
if not args.config:
_update_config_path(args)
config = Config.read_config(args.config)
if args.engine:
config.engine[
'type'] = args.engine if args.engine else 'accuracy_checker'
if 'data_source' not in config.engine:
if args.data_source is None and config.engine.type == 'data_free':
args.data_source = 'pot_dataset'
config.engine['data_source'] = args.data_source
config.configure_params(args.ac_config)
config.update_from_args(args)
if config.engine.type != 'accuracy_checker' and args.evaluate:
raise Exception(
'Can not make evaluation in simplified or data_free mode')
log_dir = _create_log_path(config)
init_logger(level=args.log_level,
file_name=os.path.join(log_dir, 'log.txt'),
progress_bar=args.pbar)
logger.info('Output log dir: {}'.format(log_dir))
metrics = optimize(config)
if metrics and logger.progress_bar_disabled:
for name, value in metrics.items():
logger.info('{: <27s}: {}'.format(name, value))
end_session_telemetry(telemetry)
def check_wrong_parametrs(argv):
parser = get_common_argument_parser()
args = parser.parse_args(args=argv)
check_dependencies(args)
if not args.config:
_update_config_path(args)
