fedlab_core.models.vgg

Module Contents

Classes

VGG

VGG model

Functions

vgg11()	VGG 11-layer model (configuration “A”)
vgg11_bn()	VGG 11-layer model (configuration “A”) with batch normalization
vgg13()	VGG 13-layer model (configuration “B”)
vgg13_bn()	VGG 13-layer model (configuration “B”) with batch normalization
vgg16()	VGG 16-layer model (configuration “D”)
vgg16_bn()	VGG 16-layer model (configuration “D”) with batch normalization
vgg19()	VGG 19-layer model (configuration “E”)
vgg19_bn()	VGG 19-layer model (configuration ‘E’) with batch normalization

class fedlab_core.models.vgg.VGG(features)

Bases: torch.nn.Module

VGG model

forward(self, x)

fedlab_core.models.vgg.vgg11()

VGG 11-layer model (configuration “A”)

fedlab_core.models.vgg.vgg11_bn()

VGG 11-layer model (configuration “A”) with batch normalization

fedlab_core.models.vgg.vgg13()

VGG 13-layer model (configuration “B”)

fedlab_core.models.vgg.vgg13_bn()

VGG 13-layer model (configuration “B”) with batch normalization

fedlab_core.models.vgg.vgg16()

VGG 16-layer model (configuration “D”)

fedlab_core.models.vgg.vgg16_bn()

VGG 16-layer model (configuration “D”) with batch normalization

fedlab_core.models.vgg.vgg19()

VGG 19-layer model (configuration “E”)

fedlab_core.models.vgg.vgg19_bn()

VGG 19-layer model (configuration ‘E’) with batch normalization