search

graykode / nlp-tutorial

Natural Language Processing Tutorial for Deep Learning Researchers
https://www.reddit.com/r/MachineLearning/comments/amfinl/project_nlptutoral_repository_who_is_studying/
MIT License
14.03k stars 3.9k forks source link

Update BERT.py #61

Open colinpeth opened 3 years ago

colinpeth commented 3 years ago

Added torch.cuda to utilize cuda if available.