Code for paper KNN-BERT: Fine-Tuning Pre-Trained Models with KNN Classifier