sacmehta / ESPNet

ESPNet: Efficient Spatial Pyramid of Dilated Convolutions for Semantic Segmentation
https://sacmehta.github.io/ESPNet/
MIT License
541 stars 112 forks source link

Amount of FLOPs for ESP module #20

Closed bloonz closed 6 years ago

bloonz commented 6 years ago

Hello @sacmehta, thank you for your work. I was wondering how much FLOPs does an ESP module require.

sacmehta commented 6 years ago

We didn’t compute FLOPs. However, it is straightforward to compute them.

msson commented 5 years ago

@bloonz @sacmehta Did you figure out how to compute FLOPs for ESPNet? I am trying but it is not that straightforward for me.

sacmehta commented 5 years ago

@msson Check this code: https://github.com/warmspringwinds/pytorch-segmentation-detection/blob/master/pytorch_segmentation_detection/utils/flops_benchmark.py