fix gpu run

atulkum · atulkum · commit f2e1ba507e2c · 2019-01-09T22:16:10.000-08:00
diff --git a/neural_ner/process_training.py b/neural_ner/process_training.py
@@ -20,9 +20,6 @@
 class Processor(object):
     def __init__(self, config, model_file_path):
         self.config = config
-
-        if torch.cuda.is_available():
-            self.config.is_cuda = True
         self.vocab = Vocab(config)
         self.model = get_model(self.vocab, config, model_file_path)
 
