Dean
Fixed a bug in the training stage where the model was not saved, commiting before training on colab
79fd7d0