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

1 vote379 views1 answer
Django queryset group by count
Python
0 vote349 views1 answer
Gensim show_topics get topic
Python
1 vote362 views1 answer
Assigning values in python
Python
0 vote360 views3 answers
Create a Tuple
Python
0 vote343 views1 answer
Print torch model python
Python
6 votes319 views5 answers
Python fill zeros left
Python
1 vote339 views1 answer
Python seek file beginning after for line in file
Python
0 vote345 views1 answer
Create python package ros 2
Python
0 vote393 views1 answer
Django admin.py date format
Python Django
0 vote357 views1 answer
Form valid django created by
Python Django