ResNet18

Contents

ResNet18#

class fl_sim.models.ResNet18(num_classes: int, pretrained: bool = False)[source]#

Bases: ResNet, CLFMixin, SizeMixin, DiffMixin

ResNet18 model for image classification.

Parameters:
  • num_classes (int) – The number of output classes.

  • pretrained (bool, default False) – If True, use a pretrained model from torchvision.models.