riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 11:11:05 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

update to hf->tf args

Browse Source
This commit is contained in:
Chris 2019-05-18 17:16:01 -04:00
parent f1433db4f1
commit 69749f3fc3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_pretrained_bert/convert_hf_checkpoint_to_tf.py
View File

@ -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)