Function run_batch_size 

async fn run_batch_size(
    model: &str,
    seq_len: u32,
    amp: bool,
    gpu: u32,
) -> Result<()>