can't find the trainable generator and encoder in the code

[522539291]

Dear author, your excellent paper has brought us a lot of inspiration. We want to train the network on a new dataset ourselves, but we can't find the trainable generator and encoder in the code, how should we retrain the network instead of downloading the pre-trained model? Looking forward to your reply, thank you.