VGG

class torchelie.models.VGG(arch: list, num_classes: int)

Construct a VGG-like model. The architecture is composed of either the number of channels or ‘M’ for a maxpool operation.

This creates a standard VGG11 with 10 classes.

add_batchnorm(remove_first=False) → torchelie.models.vgg.VGG

set_input_specs(in_channels: int) → torchelie.models.vgg.VGG

training: bool