The code for running the training process is provided in the notebook.ipynb file and the rest of the files are utility for it.