:mod:`fedlab_core.models.vgg` ============================= .. py:module:: fedlab_core.models.vgg Module Contents --------------- Classes ~~~~~~~ .. autoapisummary:: fedlab_core.models.vgg.VGG Functions ~~~~~~~~~ .. autoapisummary:: fedlab_core.models.vgg.vgg11 fedlab_core.models.vgg.vgg11_bn fedlab_core.models.vgg.vgg13 fedlab_core.models.vgg.vgg13_bn fedlab_core.models.vgg.vgg16 fedlab_core.models.vgg.vgg16_bn fedlab_core.models.vgg.vgg19 fedlab_core.models.vgg.vgg19_bn .. class:: VGG(features) Bases: :class:`torch.nn.Module` VGG model .. method:: forward(self, x) .. function:: vgg11() VGG 11-layer model (configuration "A") .. function:: vgg11_bn() VGG 11-layer model (configuration "A") with batch normalization .. function:: vgg13() VGG 13-layer model (configuration "B") .. function:: vgg13_bn() VGG 13-layer model (configuration "B") with batch normalization .. function:: vgg16() VGG 16-layer model (configuration "D") .. function:: vgg16_bn() VGG 16-layer model (configuration "D") with batch normalization .. function:: vgg19() VGG 19-layer model (configuration "E") .. function:: vgg19_bn() VGG 19-layer model (configuration 'E') with batch normalization