Skip to content

Instantly share code, notes, and snippets.

@aribornstein

aribornstein/flash text model finetune.py

Created February 24, 2021 09:18
Show Gist options
  • Save aribornstein/9b885430b31b50e481c1be5aaf5da866 to your computer and use it in GitHub Desktop.
Save aribornstein/9b885430b31b50e481c1be5aaf5da866 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
flash text model finetune.py
trainer = flash.Trainer(max_epochs = 1)
trainer.finetune(model, datamodule = datamodule)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment