Voc / vocos /train.py
Respair's picture
Upload folder using huggingface_hub
6742cf9 verified
from pytorch_lightning.cli import LightningCLI
if __name__ == "__main__":
cli = LightningCLI(run=False)
cli.trainer.fit(model=cli.model, datamodule=cli.datamodule)