training_overrides={
    "learning_rate": learning_rate,
    "num_epochs": 2,
    "seed": 123,
}

        __
        __