Update models/TextCNN.py

models/TextCNN.py

@@ -21,7 +21,8 @@ class Config(object):
         self.embedding_pretrained = torch.tensor(
             np.load(dataset + '/data/' + embedding)["embeddings"].astype('float32'))\
             if embedding != 'random' else None                                       # 向量预训练词
-        self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')   # 设备
+        #self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')   # 设备
+        self.device = torch.device('cpu')   # 设备
 
         self.dropout = 0.5                                              # 随机失活
         self.require_improvement = 1000                                 # 若超过1000batch效果还没提升，则提前结束训练