Finalizing the Training