ShrinkedResNet

Contents

ShrinkedResNet#

class fl_sim.models.ShrinkedResNet(layers: List[int], num_classes: int = 1000)[source]#

Bases: Sequential, CLFMixin, SizeMixin, DiffMixin

Shrinked version of ResNet so that the number of macro blocks can be customized (shrinked).

Parameters:

  • layers (List[int]) – Number of blocks in each macro block.

  • num_classes (int, default 1000) – The number of output classes.

Contents