incorrect model for CelebA

[euphoria0-0]

@scaldas Hi! I have a question about the model for CelebA dataset. In the paper, it said that

For the CelebA experiments, we use 10% of the total clients and the same model we described above for FEMNIST. ... For the FEMNIST experiments in the same figure, we subsample 5% of the data, and use a model with two convolutional layers followed by pooling, and a final dense layer with 2048 units.

However, the code contains 4 convolutional layers and etc. What is the correct version? Thanks.