LEARNING_RATES = [
    1e-5,
    1e-4,
    3e-4,
    1e-3,
    1e-2,
    1e-1,
]

        __
        __