How to utilize timm's scheduler?
🚀 Feature

Hi, I want to reproduce a result of image classification network by using timm library.
But I couldn't use timm.scheduler.create_scheduler because pytorch_lightning doesn't accept custom class for a scheduler.
(timm.scheduler is not the torch.optim.lr_scheduler class)
from timm.scheduler import create_scheduler
from timm.optim import create_optimizer

def configure_optimizers(self):
    optimizer = create_optimizer(self.args, self)
    scheduler, _ = create_scheduler(self.args, optimizer)

    return {
        'optimizer': optimizer,
        'lr_scheduler': scheduler,
    }

Then results this error

File "/opt/conda/lib/python3.8/site-packages/pytorch_lightning/trainer/optimizers.py", line 141, in configure_schedulers
raise ValueError(f'The provided lr scheduler "{scheduler}" is invalid')

Is there a plan for utilizing timm's scheduler?
Motivation

Pitch

Alternatives

Additional context