Any careful reader will notice that Section 3.2 of [1]} specifies that “the encoder network is trained by adding, at its output, a classification module (in all our experiments, it will be a softmax layer)”; secondly, Table 2 of [1]} clearly shows that all output sizes for the encoder configurations are different than 40. Therefore, it is necessary that the “softmax layer” must include a layer projecting the dimensionality of the encoder output to the number of classes.
