Model training configuration and checkpoints

[Zmoonzxz]

Thank you so much for open sourcing this code, but I am having trouble with training parameter configuration and testing, would you consider opening these things up?