Folders are structured as follows: dataset.py: Hyperparameters config, interface to datasets train.py: Training script eval.ipynb: IPython notebook for the evaluation of the dataset & training