Script to count Pytorch operations used by pretrained models

allenai / allennlp-hub

A collection of selected of models built with AllenNLP.

Apache License 2.0

25 stars 9 forks source link

Closed brendan-ai2 closed 4 years ago

brendan-ai2 commented 4 years ago

Counts both modules and torch.nn.functional calls.
- Limitation: Only counts the calls that are triggered by the test data.
- Also, torch.nn.functional counts are weighted by number of calls.

brendan-ai2 commented 4 years ago