eliberis / tflite-tools

TFLite model analyzer & memory optimizer
MIT License
120 stars 20 forks source link

MAC counts #5

Closed marmoi closed 2 years ago

marmoi commented 2 years ago

Hi, I found this code very useful, thanks for sharing it. While run it for my model I wanted to calculate the number of macs, and eventhough I set the parameter to True it still didn't show up anything. I made a small modification of the script "tflite_model.py" in the function "_execution_schedule_info" Before: macs = self._macs_for_op(op) if macs else 0 Modified: macs_val = self._macs_for_op(op) if macs else 0

So the variable macs does not get overwrite, and now it shows the macs count.

eliberis commented 2 years ago

Thanks for this! Fixed