CNNCifar_Tiny

class fl_sim.models.CNNCifar_Tiny(num_classes: int)

Bases: Sequential, CLFMixin, SizeMixin, DiffMixin

Convolutional neural network using CIFAR type input.

Modified from FedPD/models.py.

Input shape: (batch_size, 3, 32, 32).

Parameters:

num_classes (int) – Number of output classes.

Note

This model is too tiny.