Inconsistency with 'num_classes' vs 'num_class'

[MaggiePas]

Hey guys,

In all the module definitions the parameters are defined like so:

:param params: {
        'num_channels':1,
        'num_filters':64,
        'kernel_h':5,
        'kernel_w':5,
        'stride_conv':1,
        'pool':2,
        'stride_pool':2,
        'num_classes':28,
        'se_block': se.SELayer.None,
        'drop_out':0,2}

However in the classifier block the name of the argument for the number of classes is 'num_class':

self.conv = nn.Conv2d(
            params['num_channels'], params['num_class'], params['kernel_c'], params['stride_conv'])

Best,
Magda