Solved by Pranavan_Theivendira in post #2 
              
              
                Hi, 
from torchsummary import summary

summary(net, (3, 32, 32))

Here net(torch.nn) is your particular network and replace (3,32,32) with your image size. This gives you estimated parameter size as well as the total size. There may be other metrics. This may help you.