way better trainer

Update flow_leco_trainer.py

Create flow_leco_trainer.py

initial commit