novateur commited on
Commit
60e8581
1 Parent(s): af1973e

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/vocos_checkpoint_epoch=31_step=157696_val_loss=5.9855.ckpt
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: vocos_checkpoint_{epoch}_{step}_{val_loss:.4f}
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