Skip to content

Add profiling for PyTorch models

Javier Duarte requested to merge github/fork/AdrianAlan/feature/pytorch-profiling into master

Created by: AdrianAlan

This PR extends profiling to support Pytorch nn.Sequential models. The two new methods mirror the behavior of the methods applied to Keras models.

Further collapsing of three similar methods weights_hlsmodel, weights_keras, weights_torch (same for the activations_<type>) into one would be desirable.

Merge request reports