seperated training of segmentation stack