TencentGameMate
commited on
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 |
```
|