Explorar el Código

Use 4 workers

Lengyue hace 2 años
padre
commit
e4839d6e84
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  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}