Fix lint problems
This commit is contained in:
parent
bf00bd654a
commit
89f1f386ce
@ -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()
|
||||
|
Loading…
x
Reference in New Issue
Block a user