From a7b4cfe9194bf93c7044a42c9f1281260ce6279e Mon Sep 17 00:00:00 2001 From: carefree0910 Date: Sun, 11 Aug 2019 21:36:51 +0800 Subject: [PATCH] Update README.md I assume that it should test the `re-load` functionality after testing the `save` functionality, however I'm also surprised that nobody points this out after such a long time, so maybe I've misunderstood the purpose. This PR is just in case :) --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index b86a5238c2..48c54a055a 100644 --- a/README.md +++ b/README.md @@ -123,7 +123,7 @@ traced_model = torch.jit.trace(model, (input_ids,)) model.save_pretrained('./directory/to/save/') # save model = model_class.from_pretrained('./directory/to/save/') # re-load tokenizer.save_pretrained('./directory/to/save/') # save -tokenizer = tokenizer_class.from_pretrained(pretrained_weights) +tokenizer = tokenizer_class.from_pretrained('./directory/to/save/') # re-load # SOTA examples for GLUE, SQUAD, text generation... ```