Update models/nli.py

models/nli.py

@@ -11,7 +11,8 @@ from torch.nn import DataParallel
 from models.data_utils import build_batch, LoggingHandler, get_examples
 from tqdm import tqdm
 from sklearn.metrics import precision_recall_fscore_support
-from transformers import AutoConfig, RobertaModel, AutoModel, AutoTokenizer, AdamW
+from transformers import AutoConfig, RobertaModel, AutoModel, AutoTokenizer
+from torch.optim import AdamW
 
 import torch.nn.functional as F
 import numpy as np