diff --git a/docs/training_multilabel_ner.rst b/docs/training_multilabel_ner.rst index a7e3f8f2..d5a3c88b 100644 --- a/docs/training_multilabel_ner.rst +++ b/docs/training_multilabel_ner.rst @@ -49,6 +49,7 @@ Note, if running the script on MPS you will need to add the following to the top import os os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1" + as one of the functions used in the Transformer NER step are not supported on MPS. Our results with this approach