Fix: Correct AdamW import path

[b1n1yam]

Fix: Correct AdamW import path

Update the import statement for AdamW to use torch.optim.AdamW instead of transformers.AdamW. This corrects the import path to align with the standard PyTorch library, ensuring compatibility and proper usage of the AdamW optimizer.

What does this PR do?

This PR fixes an ImportError by correcting the import path for the AdamW optimizer in the PyTorch training code snippet. It changes the import from a potentially incorrect location (like transformers or transformers.optimization) to the correct one within the PyTorch library (torch.optim).

@Vaibhavs10
@pcuenca

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[pcuenca]

Hi @b1n1yam, the line you proposed is malfomed and would not work.

also cc @burtenshaw

[b1n1yam]

Hi @b1n1yam, the line you proposed is malfomed and would not work.

also cc @burtenshaw

Thanks for pointing that out @pcuenca - the issue has been addressed in the latest commit.

[Vaibhavs10]

thanks - this must be a very old notebook 😅

[b1n1yam]

thanks - this must be a very old notebook 😅

Yeah, it looks like this notebook was using some outdated imports. I’ve updated it to use the correct AdamW path from PyTorch. Let me know if you spot anything else that needs updating!

[pcuenca]

There's still a missing \n that needs to go at the end of line from torch.optim import AdamW

[burtenshaw]

Thanks

[pcuenca]

Thanks @burtenshaw for the final fix!