Fix CLI snippet in the docs (#12275)

This commit is contained in:
Carlos Mocholí 2022-03-16 10:28:28 +01:00 committed by GitHub
parent 7ee690758c
commit bc812077c4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions

View File

@ -891,6 +891,7 @@ If you need to customize the learning rate scheduler configuration, you can do s
.. testcode::
class MyLightningCLI(LightningCLI):
@staticmethod
def configure_optimizers(lightning_module, optimizer, lr_scheduler=None):
return ...