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 vote171 views1 answer
Getting range lowest and highest values from np array
Python
0 vote169 views1 answer
Qt line edit set text python
Python
0 vote164 views1 answer
Python 3 string length
Python
0 vote162 views1 answer
Sql o que é
Python
1 vote173 views1 answer
How to print time python 3
Python
0 vote169 views1 answer
Unshorten url python
Python
0 vote174 views1 answer
Gensim wordvector vocabulary list
Python
0 vote190 views1 answer
Mad scipy
Python
0 vote194 views1 answer
Delay print in python
Python
0 vote158 views1 answer
Check if back is pressed python
Python