CNNCifar_Small

class fl_sim.models.CNNCifar_Small(num_classes: int)

Bases: Sequential, CLFMixin, SizeMixin, DiffMixin

Convolutional neural network using CIFAR type input.

Input shape: (batch_size, 3, 32, 32), smaller in size than CNNCifar

Parameters:

num_classes (int) – Number of output classes.