Run the 'CAE_train.py' file to train the generator (encoder and decoder) and the discriminator of the CAE framework. The number of the image classes (real, fake, normal, disease1, disease2..) could be set at the "self.num_classes" parameter in the "MultiClassDis" module in "networks.py" file.