CNNCifar

Contents

CNNCifar#

class fl_sim.models.CNNCifar(num_classes: int)[source]#

Bases: Sequential, CLFMixin, SizeMixin, DiffMixin

Convolutional neural network using CIFAR type input.

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

Parameters:

num_classes (int) – Number of output classes.