Model.summary in pytorch?

vision
1

Hi. I am going over the transfer learning tutorial with ants and bees, and I am coming from Keras with no experience in Pytorch.

Is there a built-in equivalent of model.summary in pytorch?

I found this medium article but it requires installation of another framework:

Just wondering if that is indeed what I should be doing.
Thank you.

2

you can use modelsummary which is a python library ，有个python包，是可以的，github：https://github.com/graykode/modelsummary