Update wavtokenizer_smalldata_frame40_3s_nq1_code4096_dim512_kmeans200_attn.yaml
Browse files
wavtokenizer_smalldata_frame40_3s_nq1_code4096_dim512_kmeans200_attn.yaml
CHANGED
@@ -34,7 +34,7 @@ model:
|
|
34 |
|
35 |
resume: false
|
36 |
resume_config: ./WavTokenizer/configs/wavtokenizer_smalldata_frame40_3s_nq1_code16384_dim512_kmeans800_attn.yaml
|
37 |
-
resume_model: ./version_3/checkpoints/
|
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_frame40_3s_nq1_code16384_dim512_kmeans800_attn.yaml
|
37 |
+
resume_model: ./version_3/checkpoints/example.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
|