Julien Simon commited on
Commit
f4260de
1 Parent(s): 0ccc8c8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -66,7 +66,7 @@ SequenceClassifierOutput(loss=None, logits=tensor([[-0.4635, -1.0112, 4.7935,
66
  -0.1097, -1.8827, 0.6609, -0.2022, 0.0971, -0.6205, 0.4492, 0.0926,
67
  -2.4848, 0.2630, -0.4584, -2.4327, -1.1654, 0.3897, -0.3374, -1.2418,
68
  -0.1045, 0.2827, -1.5667, -0.0963]], grad_fn=<AddmmBackward0>), hidden_states=None, attentions=None)
69
- >>> classes = torch.softmax(logits.logits, dim = -1)
70
  >>> torch.set_printoptions(precision=3, sci_mode=False)
71
  >>> classes
72
  tensor([[ 0.004, 0.002, 0.701, 0.014, 0.030, 0.011,
 
66
  -0.1097, -1.8827, 0.6609, -0.2022, 0.0971, -0.6205, 0.4492, 0.0926,
67
  -2.4848, 0.2630, -0.4584, -2.4327, -1.1654, 0.3897, -0.3374, -1.2418,
68
  -0.1045, 0.2827, -1.5667, -0.0963]], grad_fn=<AddmmBackward0>), hidden_states=None, attentions=None)
69
+ >>> classes = torch.softmax(logits.logits, dim=1)
70
  >>> torch.set_printoptions(precision=3, sci_mode=False)
71
  >>> classes
72
  tensor([[ 0.004, 0.002, 0.701, 0.014, 0.030, 0.011,