Bläddra i källkod

Use 4 workers

Lengyue 2 år sedan
förälder
incheckning
e4839d6e84
1 ändrade filer med 1 tillägg och 1 borttagningar
  1. 1 1
      fish_speech/configs/llama_finetune.yaml

+ 1 - 1
fish_speech/configs/llama_finetune.yaml

@@ -41,7 +41,7 @@ data:
   _target_: fish_speech.datasets.text.TextDataModule
   train_dataset: ${train_dataset}
   val_dataset: ${val_dataset}
-  num_workers: 1
+  num_workers: 4
   batch_size: 8
   tokenizer: ${tokenizer}