Main script, loads the dataset and trains the model. Plots both the training losses and validation accuracies and returns the final test accuracy. $ python train.py -h, --help show this help message ...