SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update torchscript.rst

Browse Source 
Import fixed to pytorch_transformers else torchscript flag can't be used.
This commit is contained in:
Sukuya
2019-07-25 16:51:11 +08:00
committed by GitHub
parent 067923d326
commit 35c52f2f3c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/source/torchscript.rst
View File

@@ -74,7 +74,7 @@ according to a ``BertConfig`` class and then saved to disk under the filename ``
.. code-block:: python .. code-block:: python
from pytorch_pretrained_bert import BertModel, BertTokenizer, BertConfig from pytorch_transformers import BertModel, BertTokenizer, BertConfig
import torch import torch
enc = BertTokenizer.from_pretrained("bert-base-uncased") enc = BertTokenizer.from_pretrained("bert-base-uncased")
@@ -129,4 +129,4 @@ Using the traced model for inference is as simple as using its ``__call__`` dund
.. code-block:: python .. code-block:: python
traced_model(tokens_tensor, segments_tensors) traced_model(tokens_tensor, segments_tensors)