sksq96 / pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

sksq96/pytorch-summary Stargazers