Skip to content

GitLab

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 @@
import argparse
from ...utils import logging
from .modeling_convbert import ConvBertConfig, ConvBertModel, load_tf_weights_in_convbert
from transformers import ConvBertConfig, ConvBertModel, load_tf_weights_in_convbert
from transformers.utils import logging
logging.set_verbosity_info()
......@@ -49,4 +49,4 @@ if __name__ == "__main__":
"--pytorch_dump_path", default=None, type=str, required=True, help="Path to the output PyTorch model."
)
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