Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
eb4d7ef9
Unverified
Commit
eb4d7ef9
authored
Jul 15, 2021
by
Lysandre Debut
Committed by
GitHub
Jul 15, 2021
Browse files
Remove framework mention (#12731)
parent
959d448b
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
3 deletions
+1
-3
docs/source/serialization.rst
docs/source/serialization.rst
+1
-3
No files found.
docs/source/serialization.rst
View file @
eb4d7ef9
...
...
@@ -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.
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment