nlp-with-transformers / notebooks

Jupyter notebooks for the Natural Language Processing with Transformers book
https://transformersbook.com/
Apache License 2.0
3.85k stars 1.19k forks source link

Chapter 09: nlpaug is extremely slow (more than 60s to generate a subsituted sentence !! #68

Open jeromemassot opened 2 years ago

jeromemassot commented 2 years ago

Information

The question or comment is about chapter:

Question or comment

Has someone also noticed that the nlpaug naw.ContextualWordEmbsAug(model_path="distilbert-base-uncased", device="cuda", action="substitute") is extremely slow even on "cuda" device?

I have a decent computer Xeon E5, 64 GB, RTX Titan 24 Gbs so I cannot understand why this module is so slow.

Thanks

Best regards

Jerome