MisconfigurationException of LRScheduler for i-Code-Doc

[armunalam]

While attempting to reproduce the results using the code you kindly shared, I encountered an issue during training related to the learning rate scheduler. The traceback is as follows:
pytorch_lightning.utilities.exceptions.MisconfigurationException: The provided lr scheduler LambdaLR doesn't follow PyTorch's LRScheduler API. You should override the LightningModule.lr_scheduler_step hook with your own logic if you are using a custom LR scheduler.

This error seems to be rooted in compatibility with PyTorch Lightning's expected scheduler API.