Skip to content

GitLab

Commit 69749f3f authored by Chris's avatar Chris
Browse files

update to hf->tf args

parent f1433db4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pytorch_pretrained_bert/convert_hf_checkpoint_to_tf.py
View file @ 69749f3f
...@@ -108,4 +108,4 @@ if __name__ == "__main__": ...@@ -108,4 +108,4 @@ if __name__ == "__main__":
model = BertModel( model = BertModel(
config=BertConfig(args.config_file_path) config=BertConfig(args.config_file_path)
).from_pretrained(args.pytorch_model_name, cache_dir=args.pytorch_model_dir) ).from_pretrained(args.pytorch_model_name, cache_dir=args.pytorch_model_dir)
convert_hf_checkpoint_to_tf(model=model, ckpt_dir=args.tf_checkpoint_dir) convert_hf_checkpoint_to_tf(model=model, ckpt_dir=args.tf_checkpoint_dir)
\ No newline at end of file
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