search

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

test: Added multiple unit tests for uncovered modules #6

Closed frgfm closed 4 years ago

frgfm commented 4 years ago

This PR aims at: