训练十轮，到底看了多少数据、改了几次参数？

def plan_epoch(samples, batch_size, accumulate, drop_last=False):
    if type(samples) is not int or samples < 0:
        raise ValueError('samples must be a nonnegative integer')
    if any(type(x) is not int or x < 1 for x in (batch_size, accumulate)):
        raise ValueError('batch_size and accumulate must be positive integers')
    if type(drop_last) is not bool:
        raise ValueError('drop_last must be a boolean')

    sizes = [min(batch_size, samples - start)
             for start in range(0, samples, batch_size)]
    if drop_last and sizes and sizes[-1] < batch_size:
        sizes.pop()
    # Every remaining group is settled, including the end-of-epoch remainder.
    update_samples = [sum(sizes[start:start + accumulate])
                      for start in range(0, len(sizes), accumulate)]
    return sizes, update_samples


if __name__ == '__main__':
    for name, drop in [('keep tail', False), ('drop tail', True)]:
        batches, updates = plan_epoch(10, 4, 2, drop)
        print(f'{name}: batches={batches}, update_samples={updates}')
    batches, updates = plan_epoch(10, 4, 2)
    print(f'2 epochs, keep tail: batches={2 * len(batches)}, '
          f'planned_updates={2 * len(updates)}, sample_visits={2 * sum(batches)}')