Fixed bug in predict for cuda tensor

Author: SeanRosario

main.py

@@ -100,7 +100,7 @@ def mr(text_field, label_field, **kargs):
 
 # train or predict
 if args.predict is not None:
-    label = train.predict(args.predict, cnn, text_field, label_field)
+    label = train.predict(args.predict, cnn, text_field, label_field, args.cuda)
     print('\n[Text]  {}[Label] {}\n'.format(args.predict, label))
 elif args.test :
     try:

train.py

@@ -74,14 +74,16 @@ def eval(data_iter, model, args):
                                                                        size))
 
 
-def predict(text, model, text_field, label_feild):
+def predict(text, model, text_field, label_feild, cuda_flag):
     assert isinstance(text, str)
     model.eval()
     # text = text_field.tokenize(text)
     text = text_field.preprocess(text)
     text = [[text_field.vocab.stoi[x] for x in text]]
     x = text_field.tensor_type(text)
     x = autograd.Variable(x, volatile=True)
+    if cuda_flag:
+        x =x.cuda()
     print(x)
     output = model(x)
     _, predicted = torch.max(output, 1)