CNNFEMnist_Tiny

class fl_sim.models.CNNFEMnist_Tiny(num_classes: int = 62)

Bases: Sequential, CLFMixin, SizeMixin, DiffMixin

Tiny version of CNNFEMnist.

Modified from FedPD/models.py.

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

Parameters:

num_classes (int, default 62) – Number of output classes.