fast-bert: learner.save_model gives KeyError while saving tokenizer/vocab file

I’m trying to run the multilabel classification model and while saving the model it give me an error on vocab file learner.save_model() gives below error: image

Is this because I have not specified some path or because I’m not using a pretrained model path from local as in sample notebook.

My learner config is as below: image

DataBunchConfig as below: image

Any help appreciated. Thanks!

About this issue

  • Original URL
  • State: closed
  • Created 4 years ago
  • Comments: 17 (6 by maintainers)

Most upvoted comments

You need to pass save_model a Path to the directory you wish to save it to.