resolve AdamW deprecation warning

[rhelmeczi]

Update the optimizer usage to resolve the deprecation warning from the transformers library: "This implementation of AdamW is deprecated and will be removed in a future version. Use the PyTorch implementation torch.optim.AdamW instead, or set no_deprecation_warning=True to disable this warning"