riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

Remove framework mention (#12731)

Browse Source
This commit is contained in:
Lysandre Debut 2021-07-15 17:49:02 +02:00 committed by GitHub
parent 959d448b3f
commit eb4d7ef97b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docs/source/serialization.rst
View File

@ -72,8 +72,6 @@ The package may be used as a Python module:
-h, --help show this help message and exit
-m MODEL, --model MODEL
Model's name of path on disk to load.
-f {pytorch}, --framework {pytorch}
Framework to use when exporting. Possible values are: {'pytorch'}
--features {default} Export the model with some additional features.
--opset OPSET ONNX opset version to export the model with (default 12).
--atol ATOL Absolute difference tolerance when validating the model.
@ -82,7 +80,7 @@ Exporting a checkpoint using a ready-made configuration can be done as follows:
.. code-block::
python -m transformers.onnx -f pytorch --model=bert-base-cased onnx/bert-base-cased/
python -m transformers.onnx --model=bert-base-cased onnx/bert-base-cased/
This exports an ONNX graph of the mentioned checkpoint. Here it is `bert-base-cased`, but it can be any model from the
hub, or a local path.