can now handle single layers, for #9 - Githubissues

sksq96 / pytorch-summary

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

MIT License

4.01k stars 412 forks source link

can now handle single layers, for #9 #106

Closed Naireen closed 4 years ago