add inference example, some small improvements in training code

1. add inference example
2. some small improvements in training code