frgfm / torch-scan

Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
https://frgfm.github.io/torch-scan/latest
Apache License 2.0
208 stars 22 forks source link

Add annotation typing and mypy CI verification #29

Closed frgfm closed 3 years ago

frgfm commented 3 years ago

Considering the relatively small codebase, it would be preferable to add annotation typing early for clearer documentation and interface. A CI job should be added to enforce mypy where it's possible.