This commit is contained in:
sayakpaul
2024-12-25 21:34:56 +05:30
parent 5aef4a6c80
commit 6b54867c3c
4 changed files with 45 additions and 20 deletions
+3 -3
View File
@@ -1,14 +1,14 @@
compute_environment: LOCAL_MACHINE
debug: false
distributed_type: MULTI_GPU
distributed_type: 'NO'
downcast_bf16: 'no'
enable_cpu_affinity: false
gpu_ids: 0,1
gpu_ids: 0
machine_rank: 0
main_training_function: main
mixed_precision: bf16
num_machines: 1
num_processes: 2
num_processes: 1
rdzv_backend: static
same_network: true
tpu_env: []