multi-stream training script update