Training my own datasets

[chensongloose]

Hello! I want to use this model to train my own datasets. My data set has only four training categories.
I have the following two questions:
1.What do I need to change in the code?
2.Do I need to modify the code for the segementation and synthesis section？