Summary of our model

vision
1

how to get summary of our model?
Like in keras we have model.summary()

2

Hi @swarup

You can use:

from torchsummary import summary
model = Net() # instantiate your model here
summary(model.cuda(), (3,32,32)) # specify the input shape your model expects