training improvement

training saves periodically