# I set everything and when i'm clicking on train model, it starts but after 2 seconds there is an error