README: drop checkpoint epoch/step numbers
Browse files
README.md
CHANGED
|
@@ -19,7 +19,8 @@ reconstruction (chunk_size=4, left_context=32).
|
|
| 19 |
|
| 20 |
## Checkpoint
|
| 21 |
|
| 22 |
-
`final.pt`
|
|
|
|
| 23 |
`{epoch, hubert_state_dict, optimizer_state_dict, train_loss}`.
|
| 24 |
Load `hubert_state_dict` into
|
| 25 |
`models/model_tokenizer.py::Data2VecSemanticAcousticModel`
|
|
|
|
| 19 |
|
| 20 |
## Checkpoint
|
| 21 |
|
| 22 |
+
`final.pt` is the released tokenizer, taken from the end of the cosine
|
| 23 |
+
phase. Dict keys:
|
| 24 |
`{epoch, hubert_state_dict, optimizer_state_dict, train_loss}`.
|
| 25 |
Load `hubert_state_dict` into
|
| 26 |
`models/model_tokenizer.py::Data2VecSemanticAcousticModel`
|