amarczew / pytorch_model_summary

MIT License
52 stars 15 forks source link

'max_depth' parameter should default as None, not 1 #5

Open daniel347x opened 4 years ago

daniel347x commented 4 years ago

It was a hassle for me to get the behavior shown in the examples on the README. I was expecting to see inputs/output sizes for all of my layers, just as seen in the examples in the README. But that doesn't happen because max_depth must be set; otherwise the default of 1 results in just the top layer's inputs/outputs being shown.

Can I recommend that the default for max_depth be changed to None (so that 9999 is the result, from the current code) so that, out-of-the-box, users see all their info and they also see what is shown in the README appearing in their own output when they use the same parameters as shown in the README?