Train a model on fashion dataset