File size: 174 Bytes
6742cf9
 
 
 
 
 
1
2
3
4
5
6
7
from pytorch_lightning.cli import LightningCLI


if __name__ == "__main__":
    cli = LightningCLI(run=False)
    cli.trainer.fit(model=cli.model, datamodule=cli.datamodule)