hello, I train my data with your code with successful,but I meet a problem for i cannot how to use the model to train my data,thanks!