feat: allow more input shapes and weight locations - Githubissues

sksq96 / pytorch-summary

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

MIT License

4.01k stars 412 forks source link

feat: allow more input shapes and weight locations #189

Open ClashLuke opened 2 years ago