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

run_tf_glue MRPC evaluation only for MRPC

Browse Source
This commit is contained in:
Lysandre
2019-10-31 18:00:51 -04:00
parent be36cf92fb
commit 1a2b40cb53
1 changed files with 15 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/run_tf_glue.py
View File

@@ -71,6 +71,7 @@ history = model.fit(train_dataset, epochs=EPOCHS, steps_per_epoch=train_steps,
os.makedirs('./save/', exist_ok=True) os.makedirs('./save/', exist_ok=True)
model.save_pretrained('./save/') model.save_pretrained('./save/')
if TASK == "mrpc":
# Load the TensorFlow model in PyTorch for inspection # Load the TensorFlow model in PyTorch for inspection
pytorch_model = BertForSequenceClassification.from_pretrained('./save/', from_tf=True) pytorch_model = BertForSequenceClassification.from_pretrained('./save/', from_tf=True)