First version of the model, working well on the validation set. 02cebea tmills commited on Nov 1, 2022