Skip to content

Instantly share code, notes, and snippets.

@aribornstein

aribornstein/flash text model backbone.py

Created February 24, 2021 09:18
Show Gist options
  • Save aribornstein/33f0e41bd0279d17150419f658524fee to your computer and use it in GitHub Desktop.
Save aribornstein/33f0e41bd0279d17150419f658524fee to your computer and use it in GitHub Desktop.
Download ZIP
Raw
flash text model backbone.py
model = TextClassifier(num_classes = 2, backbone = 'roberta-base')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment