accelerate_multi_gpu.yaml 346 B

1234567891011121314151617
  1. compute_environment: LOCAL_MACHINE
  2. debug: false
  3. distributed_type: MULTI_GPU
  4. downcast_bf16: 'no'
  5. enable_cpu_affinity: false
  6. gpu_ids: all
  7. machine_rank: 0
  8. main_training_function: main
  9. mixed_precision: bf16
  10. num_machines: 1
  11. num_processes: 4
  12. rdzv_backend: static
  13. same_network: true
  14. tpu_env: []
  15. tpu_use_cluster: false
  16. tpu_use_sudo: false
  17. use_cpu: false