Draft: Add profiling features for onnx cleaner
Context
Related to aidge_onnx#83. Enhances TensorRT export module with comparative analysis of raw and cleaned models using ONNX cleaner and other frameworks.
Current limitations: external modules (aidge_onnx/onnxsim/onnxslim) should be installed in the running container for now.
Current changes
-
Profiling API: Added
getProfileData()andgetAverageExecutionTime()methods to Graph class - Enhanced test script: automated model performance comparison against multiple cleaners (onnx cleaner, onnxsim, onnxslim) with rich output
- Docker integration: Added CLI parameters to test target
Next steps - Docker containerization
Currently, aidge_onnx (and dependancies) are installed directly in the running container using a mounted aidge local repository.
TODO
-
Profiling API -
Basic compare script -
Integration strategy decision -
Clean docker installation -
Unit tests -
Documentation and examples
Edited by matthieu marchal
