search

ThilinaRajapakse / simpletransformers

Transformers for Information Retrieval, Text Classification, NER, QA, Language Modelling, Language Generation, T5, Multi-Modal, and Conversational AI
https://simpletransformers.ai/
Apache License 2.0
4.09k stars 728 forks source link

RuntimeError: NCCL Error 2: unhandled system error #1501

Open anaconda121 opened 1 year ago

anaconda121 commented 1 year ago

Describe the bug Running into a runtime error when using .predict method in simple transformers.

To Reproduce Steps to reproduce the behavior:

model = ClassificationModel("bert", "bert-base-cased")
model.predict(["this a test string"])

Expected behavior It should return a typical prediction with a class label and probability distribution.

Screenshots image

Desktop (please complete the following information):

Additional context Wondering if anyone else is having this problem.

stale[bot] commented 1 year ago

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.