Commit
·
f3c1964
1
Parent(s):
fd2c3c6
Update README.md
Browse files
README.md
CHANGED
|
@@ -51,10 +51,10 @@ input_values = input_values.to(device)
|
|
| 51 |
|
| 52 |
with torch.no_grad():
|
| 53 |
outputs = model(input_values)
|
| 54 |
-
|
| 55 |
|
| 56 |
# for Wav2Vec2ForPreTraining
|
| 57 |
# outputs = model(input_values, mask_time_indices=mask_time_indices, output_hidden_states=True)
|
| 58 |
-
#
|
| 59 |
|
| 60 |
```
|
|
|
|
| 51 |
|
| 52 |
with torch.no_grad():
|
| 53 |
outputs = model(input_values)
|
| 54 |
+
last_hidden_state = outputs.last_hidden_state
|
| 55 |
|
| 56 |
# for Wav2Vec2ForPreTraining
|
| 57 |
# outputs = model(input_values, mask_time_indices=mask_time_indices, output_hidden_states=True)
|
| 58 |
+
# last_hidden_state = outputs.hidden_states[-1]
|
| 59 |
|
| 60 |
```
|