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

3 votes135 views1 answer
Python store save data
Python
-1 vote144 views1 answer
Python launch prompt
Python
1 vote133 views2 answers
How to convert datetime to integer in python
Python
0 vote144 views1 answer
Open word from python
Python
0 vote142 views1 answer
Pss signatures python
Python
0 vote183 views1 answer
Get ip address in django
Python Django
0 vote163 views1 answer
How to put words into list
Python
12 votes145 views3 answers
How to get the url of the current page in selenium python
Python
19 votes176 views1 answer
Python ignore runtimewarning
Python
0 vote140 views1 answer
How to swirtch the placement of the levels in pandas
Python