The documentation page MAIN_CLASSES/#TRANSFORMERS.GET_COSINE_WITH_HARD_RESTARTS_SCHEDULE_WITH_WARMUP.NUM_WARMUP_STEPS doesn’t exist in v4.45.1, but exists on the main version. Click here to redirect to the main version of the documentation.