ShrinkedResNet

ShrinkedResNet#

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

基类:Sequential, CLFMixin, SizeMixin, DiffMixin

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

参数:
  • layers (List[int]) -- Number of blocks in each macro block.

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