Skip to content
Snippets Groups Projects
Commit 01071dc3 authored by Cyril Moineau's avatar Cyril Moineau
Browse files

Debug commit

parent 626a07fe
No related branches found
No related tags found
No related merge requests found
Pipeline #72788 failed
......@@ -9,8 +9,10 @@ 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"])
"lenet-no-args": ("export_LeNet/lenet.py", []),
"lenet-int8": ("export_LeNet/lenet.py", ["--dtype=int8"]),
"resnet18-no-args": ("export_ResNet18/resnet18.py", ["--mock_db"]),
"resnet18-int8": ("export_ResNet18/resnet18.py", ["--mock_db", "--dtype=int8"])
}
def generate_test_cases():
......
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment