Skip to content

Commit

Permalink
Auto chose gpus
Browse files Browse the repository at this point in the history
  • Loading branch information
leng-yue committed Dec 21, 2023
1 parent 639e2d0 commit ad6fd71
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion fish_speech/configs/vqgan_pretrain_v2.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ project: vqgan_pretrain_v2
# Lightning Trainer
trainer:
accelerator: gpu
devices: 4
devices: auto
strategy: ddp_find_unused_parameters_true
precision: 32
max_steps: 1_000_000
Expand Down

0 comments on commit ad6fd71

Please sign in to comment.