diff --git a/aidge_export_cpp/unit_tests/test_examples.py b/aidge_export_cpp/unit_tests/test_examples.py index 1ddf6590acf26b0a4f5e31b43900176551c77a7f..5d95861ddcfa2aecdf2a2851df113c05d982f084 100644 --- a/aidge_export_cpp/unit_tests/test_examples.py +++ b/aidge_export_cpp/unit_tests/test_examples.py @@ -10,7 +10,7 @@ EXAMPLES_DIR = CURRENT_DIR / "../../examples" # Dictionary of test cases: {id: (script_name, script_args)} TEST_CASES = { "lenet-int8-no-args": ("export_LeNet/lenet.py", []), - "resnet18-int8-no-args": ("export_ResNet18/resnet18.py", ["--mock_db"]) + "resnet18-int8-no-args": ("export_ResNet18/resnet18.py", ["--mock_db", "--dtype int8"]) } def generate_test_cases(): diff --git a/examples/export_ResNet18/resnet18.py b/examples/export_ResNet18/resnet18.py index 81e335567194a83794bd4be8c2dcbd8056fa3ebf..ecc2763da93e7dc4307a29f831747fb05423a054 100644 --- a/examples/export_ResNet18/resnet18.py +++ b/examples/export_ResNet18/resnet18.py @@ -274,7 +274,8 @@ Load the .onnx model and perform some usual graph modifications : # Define the target path and filename file_url = "https://huggingface.co/EclipseAidge/resnet18/resolve/main/resnet18_imagenet_1k.onnx?download=true" file_path = Path(MODEL_NAME + "_imagenet_1k.onnx") - +dir(aidge_core.utils) +aidge_core.show_version() aidge_core.utils.download_file(file_path, file_url) model = aidge_onnx.load_onnx(file_path, verbose=False)