Function get_schedule 

pub fn get_schedule(
    num_steps: usize,
    shift: Option<(usize, f64, f64)>,
) -> Vec<f64>

shift is a triple (image_seq_len, base_shift, max_shift).