Learning Rate Warmup Scheduler | TensorTonic