Move BERT to correct device when loading

NLPatVCU / medaCy

:hospital: Medical Text Mining and Information Extraction with spaCy

GNU General Public License v3.0

432 stars 91 forks source link

Closed ggurdin closed 4 years ago

ggurdin commented 4 years ago

Added line to BERT learner to move self.model to correct device when loading