Get rid of deprecation warning with AdamW optimiser

scripts/ML/BERT/Trainer.py

@@ -3,7 +3,8 @@ import datetime
 from loaders import set_random
 import time
 import torch
-from transformers import AdamW, get_linear_schedule_with_warmup
+from torch.optim import AdamW
+from transformers import get_linear_schedule_with_warmup
 
 def chrono(f):
     def wrapped(*args, **kwargs):