feat: Added module support

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

Closed frgfm closed 4 years ago

frgfm commented 4 years ago

This PR aims at :

adding ops estimation support for torch.nn.Identity and Transposed convolutions
adding corresponding tests