SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[run_tf_glue] Add comment for context

Browse Source
This commit is contained in:
Julien Chaumond
2019-11-05 19:56:43 -05:00
parent 30968d70af
commit f88c104d8f
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/run_tf_glue.py
View File

@@ -73,6 +73,8 @@ model.save_pretrained('./save/')
if TASK == "mrpc": if TASK == "mrpc":
# Load the TensorFlow model in PyTorch for inspection # Load the TensorFlow model in PyTorch for inspection
# This is to demo the interoperability between the two frameworks, you don't have to
# do this in real life (you can run the inference on the TF model).
pytorch_model = BertForSequenceClassification.from_pretrained('./save/', from_tf=True) pytorch_model = BertForSequenceClassification.from_pretrained('./save/', from_tf=True)
# Quickly test a few predictions - MRPC is a paraphrasing task, let's see if our model learned the task # Quickly test a few predictions - MRPC is a paraphrasing task, let's see if our model learned the task