Download FFHQ dataset and unzip it to the /data/ffhq directory.
Download age label to the /data directory.
You can also train the model with your own dataset. Put your images in the /data directory. Modify the dataset path in train.py. With the pretrained classifier, you can create a new label file with the age of each image.