Train from Scratch

Make a totally new model

You can start using the pretrained model here: https://huggingface.co/Blackroot/Anime-Aesthetic-Predictor-Medium/tree/main or you can use the create fresh classification model notebook to make an ENTIRELY NEW network that needs to be retrained on a much larger number of images.

Run the create_fresh_classifcation_model.ipynb notebook.

Run the entire notebook, it will create a new untrained classifcation network based on ConVnext automatically called "full_model.pth"

From here, the training process is the same, it just requires substantially more images and time.