Update train_utils.py

Author: CurryTang

train_utils.py

@@ -50,6 +50,7 @@ def get_optimizer(args, model):
 
 
 def train(model, data, optimizer, loss_fn, train_mask, val_mask):
+    model.train()
     optimizer.zero_grad()
     preds = model(data)
     if len(data.y.shape) != 1:
@@ -437,4 +438,4 @@ def annotator(pred_texts, label_names):
             conf.append(0)
 
     anno = torch.LongTensor(anno)
-    return anno, conf
+    return anno, conf