The AdamW function has moved from the transformers library, it now needs to be imported as follows:
from torch.optim import AdamW
2 Likes
The AdamW function has moved from the transformers library, it now needs to be imported as follows:
from torch.optim import AdamW