search

utterworks / fast-bert

Super easy library for BERT based NLP models
Apache License 2.0
1.85k stars 342 forks source link

Allow setting custom device in prediction.py #251

Closed ddelange closed 3 years ago

ddelange commented 3 years ago

Hi :)

This allows using custom torch devices to predict using a saved model.

ddelange commented 3 years ago

@kaushaltrivedi any chance of merging (if noting else needed here)?

kaushaltrivedi commented 3 years ago

done.