NielsRogge / Transformers-Tutorials

This repository contains demos I made with the Transformers library by HuggingFace.
MIT License
8.51k stars 1.34k forks source link

Pix2Struct: The provided lr scheduler `LambdaLR` doesn't follow PyTorch's LRScheduler API #291

Closed JakeRutherford closed 1 year ago

JakeRutherford commented 1 year ago

I executed the Pix2Struct notebook as is, and then got this error:

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.

When running trainer.fit(pl_module)