Showsummary

In [ ]:

from torchsummary import summary
from models import LeNet,AlexNet,ResNet,VGGNet,MobileNet,GoogLeNet

from torchsummary import summary from models import LeNet,AlexNet,ResNet,VGGNet,MobileNet,GoogLeNet

In [ ]:

print('######LeNet######')
model = LeNet.LeNet()
summary(model, (1,32,32), device='cpu')

print('######LeNet######') model = LeNet.LeNet() summary(model, (1,32,32), device='cpu')

In [ ]:

print('######AlexNet######')
model = AlexNet.AlexNet()
summary(model, (1,227,227), device='cpu')

print('######AlexNet######') model = AlexNet.AlexNet() summary(model, (1,227,227), device='cpu')

In [ ]:

print('######ResNet######')
model = ResNet.ResNet()
summary(model, (1,224,224), device='cpu')

print('######ResNet######') model = ResNet.ResNet() summary(model, (1,224,224), device='cpu')

In [ ]:

print('######VGGNet######')
model = VGGNet.VGGNet()
summary(model, (1,64,64), device='cpu')

print('######VGGNet######') model = VGGNet.VGGNet() summary(model, (1,64,64), device='cpu')

In [ ]:

print('######MobileNet######')
model = MobileNet.MobileNet()
summary(model, (1,128,128), device='cpu')

print('######MobileNet######') model = MobileNet.MobileNet() summary(model, (1,128,128), device='cpu')

In [ ]:

print('######GoogLeNet######')
model = GoogLeNet.GoogLeNet()
summary(model, (1,128,128), device='cpu')

print('######GoogLeNet######') model = GoogLeNet.GoogLeNet() summary(model, (1,128,128), device='cpu')

In [ ]:

'''
print('######VGGNet######')
model = VGGNet.VGGNet()
summary(model, (1,128,128), device='cpu')
'''

''' print('######VGGNet######') model = VGGNet.VGGNet() summary(model, (1,128,128), device='cpu') '''