Skip to content

Instantly share code, notes, and snippets.

Last active Jul 16, 2020
What would you like to do?
from simpletransformers.classification import ClassificationModel
# set use_cuda=False on CPU-only platforms
model = ClassificationModel('bert', 'monsoon-nlp/dv-wave', num_labels=8, use_cuda=True, args={
'reprocess_input_data': True,
'use_cached_eval_features': False,
'overwrite_output_dir': True,
'num_train_epochs': 3,
'silent': True
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment