Hello, I'm trying to implement the training part, if anyone also want to implement this. Just contact with me. Thanks.