minor change to multi-stream training script