Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "d4564df1d4a6b355779f1a8ac250cb47cb4c38d8"
Unverified Commit aa438a42 authored by Stefan Schweter's avatar Stefan Schweter Committed by GitHub
Browse files

convbert: minor fixes for conversion script (#9937)

parent 62024453
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
src/transformers/models/convbert/convert_convbert_original_tf1_checkpoint_to_pytorch.py
View file @ aa438a42
...@@ -16,8 +16,8 @@ ...@@ -16,8 +16,8 @@
import argparse import argparse
from ...utils import logging from transformers import ConvBertConfig, ConvBertModel, load_tf_weights_in_convbert
from .modeling_convbert import ConvBertConfig, ConvBertModel, load_tf_weights_in_convbert from transformers.utils import logging
logging.set_verbosity_info() logging.set_verbosity_info()
...@@ -49,4 +49,4 @@ if __name__ == "__main__": ...@@ -49,4 +49,4 @@ if __name__ == "__main__":
"--pytorch_dump_path", default=None, type=str, required=True, help="Path to the output PyTorch model." "--pytorch_dump_path", default=None, type=str, required=True, help="Path to the output PyTorch model."
) )
args = parser.parse_args() args = parser.parse_args()
convert_orig_tf1_checkpoint_to_pytorch(args.tf_checkpoint_path, args.conv_bert_config_file, args.pytorch_dump_path) convert_orig_tf1_checkpoint_to_pytorch(args.tf_checkpoint_path, args.convbert_config_file, args.pytorch_dump_path)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment