First posted on my Kaggle.
BREAKING NEWS: We are making a news topic classifier!
Here we'll be doing transfer learning on BERT with the AG News Dataset in PyTorch.
To try out the model without running the notebook, check out my inference endpoint at HuggingFace.