train

Thank you dear, for awesome project. Can we chance to train the model (based on human face,face attributes) using diffusemix augmentaiton? Where is the train script?