Update wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml
Browse files
wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml
CHANGED
|
@@ -34,7 +34,7 @@ model:
|
|
| 34 |
|
| 35 |
resume: false
|
| 36 |
resume_config: ./WavTokenizer/configs/wavtokenizer_smalldata_frame75_3s_nq1_code16384_dim512_kmeans800_attn.yaml
|
| 37 |
-
resume_model: ./WavTokenizer/result/train/wavtokenizer_smalldata_frame75_3s_nq1_code16384_dim512_kmeans800_attn/
|
| 38 |
|
| 39 |
feature_extractor:
|
| 40 |
class_path: decoder.feature_extractors.EncodecFeatures
|
|
@@ -77,7 +77,7 @@ trainer:
|
|
| 77 |
- class_path: pytorch_lightning.callbacks.ModelCheckpoint
|
| 78 |
init_args:
|
| 79 |
monitor: val_loss
|
| 80 |
-
filename:
|
| 81 |
save_top_k: 10
|
| 82 |
save_last: true
|
| 83 |
- class_path: decoder.helpers.GradNormCallback
|
|
|
|
| 34 |
|
| 35 |
resume: false
|
| 36 |
resume_config: ./WavTokenizer/configs/wavtokenizer_smalldata_frame75_3s_nq1_code16384_dim512_kmeans800_attn.yaml
|
| 37 |
+
resume_model: ./WavTokenizer/result/train/wavtokenizer_smalldata_frame75_3s_nq1_code16384_dim512_kmeans800_attn/xxx.ckpt
|
| 38 |
|
| 39 |
feature_extractor:
|
| 40 |
class_path: decoder.feature_extractors.EncodecFeatures
|
|
|
|
| 77 |
- class_path: pytorch_lightning.callbacks.ModelCheckpoint
|
| 78 |
init_args:
|
| 79 |
monitor: val_loss
|
| 80 |
+
filename: wavtokenizer_checkpoint_{epoch}_{step}_{val_loss:.4f}
|
| 81 |
save_top_k: 10
|
| 82 |
save_last: true
|
| 83 |
- class_path: decoder.helpers.GradNormCallback
|