batch_size: <list[int]>
A list of potential batch sizes during data loading.
batch_size
<=2048
[512, 1024]