TorchScan: inspect your PyTorch models#
The torchscan
package provides tools for analyzing your PyTorch modules and models. Additionally to performance benchmarks, a comprehensive architecture comparison require some insights in the model complexity, its usage of computational and memory resources.
This project is meant for:
⚡ exploration: easily assess the influence of your architecture on resource consumption
👩🔬 research: quickly implement your own ideas to mitigate latency
Supported layers#
Here is the list of supported layers for FLOPS, MACs, DMAs and receptive field computation:
Non-linear activations#
Linear layers#
Convolutions#
Pooling#
Normalization#
Other#
Please note that the functional API of PyTorch is not supported.