Skip to content

Instantly share code, notes, and snippets.

@aribornstein

aribornstein/checkpoint flash text model.py

Created February 24, 2021 09:22
Show Gist options
  • Save aribornstein/bc48c1509def24470f70ea1b20a8d731 to your computer and use it in GitHub Desktop.
Save aribornstein/bc48c1509def24470f70ea1b20a8d731 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
checkpoint flash text model.py
# Save Checkpoint
trainer.save_checkpoint("text_class_model.pt")
# Load Model From Checkpoint
model = TextClassifier.load_from_checkpoint("text_class_model.pt")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment