Pytorch summary model

from torchsummary import summary
help(summary)
import torchvision.models as models
alexnet = models.alexnet(pretrained=False)
alexnet.cuda()
summary(alexnet, (3, 224, 224))
print(alexnet)

Source: stackoverflow.com

Similar codes

0 vote216 views1 answer
Build the union of a list of RDDs
Python
2 votes189 views2 answers
How to replace a single letter in a string python
Python
0 vote185 views1 answer
Hide password input tkinter
Python
1 vote230 views1 answer
How to scroll by in selenium python
Python
0 vote192 views1 answer
Bitbucket rest api python example
Python
0 vote229 views1 answer
Python package for misspelled words
Python
1 vote200 views1 answer
How to draw shape square in python turtle
Python
0 vote224 views1 answer
Turtle meaning
Python
0 vote200 views1 answer
Micropython string to int
Python
1 vote240 views1 answer
How to add char to string python
Python