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 vote159 views1 answer
Python tf.maximum
Python
0 vote201 views1 answer
Aggregate the elements of each partition, and then the results for all the partitions
Python
0 vote160 views1 answer
Python how to use rnage
Python
0 vote193 views1 answer
String times python
Python
0 vote160 views1 answer
Zeromq pub sub example python
Python
0 vote179 views1 answer
Python read column from csv
Python
2 votes181 views2 answers
Conda create jupyter kernel
Python
0 vote175 views1 answer
Serializing nested objects drf,django restframework nested object serialization
Python
0 vote176 views1 answer
{% load humanise %}
Python
0 vote184 views1 answer
Python split string after substring
Python