details of model_dir

• when training is finished, several files will be generated in the model_dir directory

  file	description
  arguments.json	the arguments for training
  loss.csv	the training records
  model.pth	the PyTorch model file, generated by torch.save()
  tags.json	the tag set
  vocab.json	the vocabulary

• only three files(model.pth, tags.json, vocab.json) are needed when predicting

  • you can remove arguments.json and loss.csv before you release your model