Remove training data after model is ready, which massively reduces pickled size