Hi!
Are you going to add a training script? It's very interesting.