diff --git a/medacy/pipelines/lstm_systematic_review_pipeline.py b/medacy/pipelines/lstm_systematic_review_pipeline.py index cfd1963c..38769352 100644 --- a/medacy/pipelines/lstm_systematic_review_pipeline.py +++ b/medacy/pipelines/lstm_systematic_review_pipeline.py @@ -36,7 +36,7 @@ def __init__(self, entities, metamap=None, **kwargs): raise ValueError('This pipeline requires word embeddings.') self.word_embeddings = kwargs['word_embeddings'] - self.cuda_device = kwargs['cuda_device'] + self.cuda_device = kwargs.get('cuda_device', -1) def get_learner(self): learner = BiLstmCrfLearner(self.word_embeddings, self.cuda_device)