CNNMnist

class fl_sim.models.CNNMnist(num_classes: int)

Bases: Sequential, CLFMixin, SizeMixin, DiffMixin

Convolutional neural network using MNIST type input.

Modified from FedPD/models.py.

Input size: (batch_size, 1, 28, 28).

Parameters:

num_classes (int) – Number of output classes.