Update README.md
Lines 183 - 200, fixed indentation. Line 198, replaced `tokenizer_class` with `BertTokenizer`, since `tokenizer_class` is not defined in the loop it belongs to.
This commit is contained in:
committed by
Lysandre Debut
parent
2dc8cb8734
commit
5c3b32d44d
@@ -195,7 +195,7 @@ traced_model = torch.jit.trace(model, (input_ids,))
|
|||||||
model.save_pretrained('./directory/to/save/') # save
|
model.save_pretrained('./directory/to/save/') # save
|
||||||
model = model_class.from_pretrained('./directory/to/save/') # re-load
|
model = model_class.from_pretrained('./directory/to/save/') # re-load
|
||||||
tokenizer.save_pretrained('./directory/to/save/') # save
|
tokenizer.save_pretrained('./directory/to/save/') # save
|
||||||
tokenizer = tokenizer_class.from_pretrained('./directory/to/save/') # re-load
|
tokenizer = BertTokenizer.from_pretrained('./directory/to/save/') # re-load
|
||||||
|
|
||||||
# SOTA examples for GLUE, SQUAD, text generation...
|
# SOTA examples for GLUE, SQUAD, text generation...
|
||||||
```
|
```
|
||||||
|
|||||||
Reference in New Issue
Block a user