Skip to content

Commit

Permalink
Update the threshold while generating optimized model file. Previousl…
Browse files Browse the repository at this point in the history
…y, it creates empty file is no initialize exceeds the threshold.
  • Loading branch information
HectorSVC committed Jan 27, 2025
1 parent 5e0598d commit 403ded3
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions onnxruntime/test/python/onnxruntime_test_python.py
Original file line number Diff line number Diff line change
Expand Up @@ -183,7 +183,7 @@ def test_model_serialization_with_original_external_initializers_to_directory(se
so.add_session_config_entry(
"session.optimized_model_external_initializers_file_name", external_initializers_file
)
so.add_session_config_entry("session.optimized_model_external_initializers_min_size_in_bytes", "100")
so.add_session_config_entry("session.optimized_model_external_initializers_min_size_in_bytes", "20")
onnxrt.InferenceSession(get_name("model_with_orig_ext_data.onnx"), sess_options=so)
self.assertTrue(os.path.isfile(so.optimized_model_filepath))
self.assertTrue(os.path.isfile(os.path.join(directory, external_initializers_file)))
Expand Down Expand Up @@ -213,14 +213,10 @@ def test_model_serialization_with_original_external_initializers_to_current_dire
"session.optimized_model_external_initializers_file_name", external_initializers_file
)

# TODO(anyone): Set this to 100 will cause test error since some tensor below the threshold
# still refers to the original external data file. We shall fix this issue so that the
# optimized model only refers to one external data file.
so.add_session_config_entry("session.optimized_model_external_initializers_min_size_in_bytes", "10")
so.add_session_config_entry("session.optimized_model_external_initializers_min_size_in_bytes", "100")
session1 = onnxrt.InferenceSession(get_name("model_with_orig_ext_data.onnx"), sess_options=so)
del session1
self.assertTrue(os.path.isfile(optimized_model_filepath))
self.assertTrue(os.path.isfile(external_initializers_file))

so2 = onnxrt.SessionOptions()
so2.log_severity_level = 1
Expand Down

0 comments on commit 403ded3

Please sign in to comment.