소스 검색

默认关闭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