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
"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "ac52084bf29ab02ee4cbc79d7330562b5df17df2"
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:
...
@@ -72,8 +72,6 @@ The package may be used as a Python module:
-
h
,
--
help
show
this
help
message
and
exit
-
h
,
--
help
show
this
help
message
and
exit
-
m
MODEL
,
--
model
MODEL
-
m
MODEL
,
--
model
MODEL
Model
's name of path on disk to load.
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.
--features {default} Export the model with some additional features.
--opset OPSET ONNX opset version to export the model with (default 12).
--opset OPSET ONNX opset version to export the model with (default 12).
--atol ATOL Absolute difference tolerance when validating the model.
--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:
...
@@ -82,7 +80,7 @@ Exporting a checkpoint using a ready-made configuration can be done as follows:
.. code-block::
.. 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
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.
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