Update default external_data_location for pre-process of quantization (#16399)

external_data_location should be a string/file_name to indicate the file
name of external data instead of a directory
This commit is contained in:
Yufeng Li 2023-06-20 09:37:17 -07:00 committed by GitHub
parent b8917ad84f
commit d190db7fcd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions

View file

@ -101,7 +101,7 @@ Essentially this tool performs the following three (skippable) steps:
parser.add_argument(
"--external_data_location",
help="The file location to save the external file",
default="./",
default=None,
)
parser.add_argument(
"--external_data_size_threshold",

View file

@ -32,7 +32,7 @@ def quant_pre_process(
verbose: int = 0,
save_as_external_data: bool = False,
all_tensors_to_one_file: bool = False,
external_data_location: str = "./",
external_data_location: str = None,
external_data_size_threshold: int = 1024,
) -> None:
"""Shape inference and model optimization, in preparation for quantization.