added recovery of training progress
fixed bug