Oh my god, thank you very much!
But what is the calculation to find that value directly from the code? Because in the code the scheduler and optimizer are instantiated first and then the Trainer.
How do I get that information fo steps before instantiating the Trainer to pass it as a parameter of get_linear_schedule_with_warmup?