allenai / papermage

library supporting NLP and CV research on scientific papers
https://papermage.org
Apache License 2.0
692 stars 54 forks source link

`TestEntityClassificationPredictorTrainer` fails on Mac M1 chip #12

Open kyleclo opened 1 year ago

kyleclo commented 1 year ago
FAILED tests/test_trainers/test_entity_classification_predictor_trainer.py::TestEntityClassificationPredictorTrainer::test_train - RuntimeError: Placeholder storage has not been allocated on MPS device!

googling, looks like it's an M1 chip issue. need to modify statement where send .to() in torch:

https://stackoverflow.com/questions/74724120/pytorch-on-m1-mac-runtimeerror-placeholder-storage-has-not-been-allocated-on-m

Not that high priority