experiment_grid = [
    (batch_size, learning_rate)
    for batch_size in BATCH_SIZES
    for learning_rate in LEARNING_RATES
]

        __
        __