Model.summary in pytorch?

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.

you can use modelsummary which is a python library ,有个python包,是可以的,github: