ResNet18

class fl_sim.models.ResNet18(num_classes: int, pretrained: bool = False)

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.