Skip to content

Commit

Permalink
update docu #3
Browse files Browse the repository at this point in the history
  • Loading branch information
nprouvost committed Feb 6, 2024
1 parent aa7b122 commit f9ee8c7
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -83,8 +83,8 @@ law run PlotRuntimes --version test_mlprof --print-output 0
## Profiling

This tools uses the c++ `<chrono>` 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
Expand Down

0 comments on commit f9ee8c7

Please sign in to comment.