After that I want to give the user an easy interface for training their own models (transfer learning using an existing base model).