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:

  
      
      Towards Data Science – 27 Sep 18
  
  
    

Model summary in PyTorch 3

Keras style model.summary() in PyTorch

  Reading time: 3 min read
    

  
  
    
    
  
  


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