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 vote201 views1 answer
Id() python
Python
0 vote199 views1 answer
Tkinter sin
Python
19 votes242 views2 answers
Set background colour tkinter
Python
0 vote221 views1 answer
Df = df.apply(lambda x: np.square(x) if x.name in [ 'a', 'e', 'g'] else x, axis=1)
Python
38 votes189 views3 answers
Matplotlib window size
Python
0 vote214 views1 answer
Natsort python pip install
Python
0 vote202 views1 answer
Oop - Apa metaclasses di Python
Python
0 vote219 views1 answer
H2o dataframe columns drop
Python
0 vote206 views1 answer
Jupyter nbconvert
Python
60 votes285 views6 answers
Python current date
Python