Explorar o código

Use 4 workers

Lengyue %!s(int64=2) %!d(string=hai) anos
pai
achega
e4839d6e84
Modificáronse 1 ficheiros con 1 adicións e 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}