frgfm / torch-scan

Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)

Home Page:https://frgfm.github.io/torch-scan/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

frgfm/torch-scan Stargazers