fix loading from tf/pt

This commit is contained in:
thomwolf
2019-09-25 17:46:16 +02:00
parent a049c8043b
commit 3b7fb48c3b
3 changed files with 5 additions and 4 deletions

View File

@@ -27,7 +27,8 @@ tf_model.save_pretrained('./runs/')
pt_model = BertForSequenceClassification.from_pretrained('./runs/')
# Quickly inspect a few predictions
inputs = tokenizer.encode_plus("I said the company is doing great", "The company has good results", add_special_tokens=True)
pred = pt_model(torch.tensor([tokens]))
# Divers
import torch