torchfl.models.sota package

Submodules

torchfl.models.sota.alexnet module

torchfl.models.sota.densenet module

torchfl.models.sota.lenet module

torchfl.models.sota.mlp module

torchfl.models.sota.mobilenet module

torchfl.models.sota.resnet module

torchfl.models.sota.shufflenetv2 module

torchfl.models.sota.squeezenet module

torchfl.models.sota.vgg module

Module contents

Sub-package for state-of-the-art model implementations pre-trained by torchvision and used by the core models provided by torchfl.

Available modules:

• alexnet

• resnet

• vgg

• squeezenet

• densenet

• shufflenetv2

• mobilenet