# Train model with lightning
# To be continued...