Преглед на файлове

默认关闭cudnn benchmark (#182)

* disable cudnn benchmark default

* disable cudnn benchmark default

* disable cudnn benchmark default
Stardust·减 преди 2 години
родител
ревизия
f767bae895
променени са 2 файла, в които са добавени 2 реда и са изтрити 0 реда
  1. 1 0
      fish_speech/configs/vits_decoder_finetune.yaml
  2. 1 0
      fish_speech/configs/vits_decoder_pretrain.yaml

+ 1 - 0
fish_speech/configs/vits_decoder_finetune.yaml

@@ -15,6 +15,7 @@ trainer:
   precision: 32
   max_steps: 100_000
   val_check_interval: 100
+  benchmark: false
 
 sample_rate: 44100
 hop_length: 512

+ 1 - 0
fish_speech/configs/vits_decoder_pretrain.yaml

@@ -14,6 +14,7 @@ trainer:
   precision: 32
   max_steps: 1_000_000
   val_check_interval: 1000
+  benchmark: false
 
 sample_rate: 44100
 hop_length: 512