composer.models.resnets#

composer.models.resnets

Classes

CIFAR_ResNet

A residual neural network as originally designed for CIFAR-10.

Initializer

An enumeration.

Attributes

  • List

  • Tuple

class composer.models.resnets.CIFAR_ResNet(plan, initializers, outputs=10)[source]#

Bases: torch.nn.modules.module.Module

A residual neural network as originally designed for CIFAR-10.

class Block(f_in, f_out, downsample=False)[source]#

Bases: torch.nn.modules.module.Module

A ResNet block.

static get_model_from_name(model_name, initializers, outputs=10)[source]#

The naming scheme for a ResNet is ‘cifar_resnet_D[_W]’.

D is the model depth (e.g. cifar_resnet56)