Fix lint problems

This commit is contained in:
William D. Irons 2020-01-11 23:59:21 +00:00
parent bf00bd654a
commit 89f1f386ce

View File

@ -753,7 +753,8 @@ def convert_with_tensorrt(args):
max_workspace_size_bytes=args.max_workspace_size_bytes,
precision_mode=args.precision_mode,
minimum_segment_size=args.minimum_segment_size)
converter = trt.TrtGraphConverterV2(input_saved_model_dir=args.dir,
converter = trt.TrtGraphConverterV2(
input_saved_model_dir=args.dir,
input_saved_model_tags=args.tag_set.split(','),
conversion_params=params)
converter.convert()