Prompsit
/

paraphrase-bert-en

Text Classification

Model card Files Files and versions

mplaza commited on Dec 14, 2021

Commit

082ded0

·

1 Parent(s): de09f41

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -37,7 +37,7 @@ logits = model(**input).logits
 soft = torch.nn.Softmax(dim=1)
 print(soft(logits))
 ```
-Output of previous code is:
  ```
  tensor([[0.1592, 0.8408]], grad_fn=<SoftmaxBackward>)
  ```

 soft = torch.nn.Softmax(dim=1)
 print(soft(logits))
 ```
+Code output is:
  ```
  tensor([[0.1592, 0.8408]], grad_fn=<SoftmaxBackward>)
  ```