Lengyue 1 год назад
Родитель
Сommit
841c8902f9
1 измененных файлов с 5 добавлено и 1 удалено
  1. 5 1
      fish_speech/configs/vits_decoder.yaml

+ 5 - 1
fish_speech/configs/vits_decoder.yaml

@@ -13,7 +13,7 @@ trainer:
   strategy: ddp_find_unused_parameters_true
   precision: 32
   max_steps: 1_000_000
-  val_check_interval: 2000
+  val_check_interval: 1000
 
 sample_rate: 44100
 hop_length: 512
@@ -117,3 +117,7 @@ callbacks:
     sub_module: 
       - generator
       - discriminator
+
+  model_checkpoint:
+    every_n_train_steps: 1000
+    save_top_k: 10