training refinement stages
saving output of training into output directory