Views
No views yet
1VGG.vgg11()
2VGG.vgg13()
3VGG.vgg16()
4VGG.vgg19()
5VGG.vgg11_bn()
6VGG.vgg13_bn()
7VGG.vgg16_bn()
8VGG.vgg19_bn()1# change activation
2VGG.vgg11(activation = nn.SELU)
3# change number of classes (default is 1000 )
4VGG.vgg11(n_classes=100)
5# pass a different block
6from nn.models.classification.senet import SENetBasicBlock
7VGG.vgg11(block=SENetBasicBlock)
8# store the features tensor after every block