Added a bigger Kryo serialization buffer, and also serialized the training data.