From f9ee8c7d1dc1e3791fd547a905779ae8fb12ffd2 Mon Sep 17 00:00:00 2001 From: Nathan Prouvost Date: Tue, 6 Feb 2024 16:56:13 +0100 Subject: [PATCH] update docu #3 --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 9465fa8..7adbd39 100644 --- a/README.md +++ b/README.md @@ -83,8 +83,8 @@ law run PlotRuntimes --version test_mlprof --print-output 0 ## Profiling This tools uses the c++ `` library for runtime measurements and (soon) [IgProf](https://igprof.org/) for the memory profiling. -It allows for the measurement of TensorFlow graphs with several input layers. -As this tool is set to work in CMSSW, it requires a frozen graph (it is recommended to use the cmsml [save_graph](https://cmsml.readthedocs.io/en/latest/api/tensorflow.html#cmsml.tensorflow.save_graph) function with the argument "True" for variables_to_constant). +It allows for the measurement of TensorFlow graphs (.pb) and ONNX models (.onnx) with several input layers. +As this tool is set to work in CMSSW, it requires a frozen graph for TensorFlow models (it is recommended to use the cmsml [save_graph](https://cmsml.readthedocs.io/en/latest/api/tensorflow.html#cmsml.tensorflow.save_graph) function with the argument "True" for variables_to_constant). ## Runtime measurement