Torchscan documentation

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.

Package Reference

• torchscan
• torchscan.modules
• torchscan.process
• torchscan.utils

Indices and tables

• Index

• Module Index