Update README.md
Browse files
README.md
CHANGED
@@ -37,7 +37,7 @@ logits = model(**input).logits
|
|
37 |
soft = torch.nn.Softmax(dim=1)
|
38 |
print(soft(logits))
|
39 |
```
|
40 |
-
|
41 |
```
|
42 |
tensor([[0.1592, 0.8408]], grad_fn=<SoftmaxBackward>)
|
43 |
```
|
|
|
37 |
soft = torch.nn.Softmax(dim=1)
|
38 |
print(soft(logits))
|
39 |
```
|
40 |
+
Code output is:
|
41 |
```
|
42 |
tensor([[0.1592, 0.8408]], grad_fn=<SoftmaxBackward>)
|
43 |
```
|