Przeglądaj źródła

Auto chose gpus

Lengyue 2 lat temu
rodzic
commit
ad6fd717ab
1 zmienionych plików z 1 dodań i 1 usunięć
  1. 1 1
      fish_speech/configs/vqgan_pretrain_v2.yaml

+ 1 - 1
fish_speech/configs/vqgan_pretrain_v2.yaml

@@ -7,7 +7,7 @@ project: vqgan_pretrain_v2
 # Lightning Trainer
 # Lightning Trainer
 trainer:
 trainer:
   accelerator: gpu
   accelerator: gpu
-  devices: 4
+  devices: auto
   strategy: ddp_find_unused_parameters_true
   strategy: ddp_find_unused_parameters_true
   precision: 32
   precision: 32
   max_steps: 1_000_000
   max_steps: 1_000_000