Script for training BERT model doesn't save the model...

[illided]

Script run_finetune_kaznerd.py doesn't actually save the model, which makes it impossible to use for inference. Can be fixed with simple trainer.save(model_path) right after the trainer.train()