Updated docs for generate_training