def get_iteration(nb, batch_size):
    """
    Given total sample size and batch size, return
    number of batch.
    """
    basic = nb//batch_size
    total = basic + (0 if nb%batch_size==0 else 1)
    return  total