ResNet18

目录

ResNet18#

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

基类:ResNet, CLFMixin, SizeMixin, DiffMixin

ResNet18 model for image classification.

参数:
  • num_classes (int) -- The number of output classes.

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