CNNCifar_Small

CNNCifar_Small#

class fl_sim.models.CNNCifar_Small(num_classes: int)[源代码]#

基类:Sequential, CLFMixin, SizeMixin, DiffMixin

Convolutional neural network using CIFAR type input.

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

参数:

num_classes (int) -- Number of output classes.