search

cezbloch / pytorchures

Python profiler for pytorch models. Wraps each layer and measures it's execution time.
MIT License
2 stars 0 forks source link

Excessive measurements visible in logs #14

Open cezbloch opened 1 month ago

cezbloch commented 1 month ago

Probably if consecutive layers have same layer type stacked they are logged under same dictionary key in the results.

cezbloch commented 4 weeks ago

when debugging the code it turns out the layer is actually called a few times for one inference run. This happens for 'retinanet_resnet50_fpn' model. It does not happen in 'ssdlite320_mobilenet_v3_large' model though.