BATCH_SIZES = [2, 4, 8]