-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathacc.yaml
40 lines (39 loc) · 929 Bytes
/
acc.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
compute_environment: LOCAL_MACHINE
debug: false
distributed_type: MULTI_GPU
downcast_bf16: 'no'
machine_rank: 0
main_training_function: main
mixed_precision: bf16
num_machines: 1
num_processes: 2
rdzv_backend: static
same_network: true
tpu_env: []
tpu_use_cluster: false
tpu_use_sudo: false
use_cpu: false
#compute_environment: LOCAL_MACHINE
#deepspeed_config:
# deepspeed_multinode_launcher: standard
# gradient_accumulation_steps: 1
# gradient_clipping: 1.0
# offload_optimizer_device: none
# offload_param_device: none
# zero3_init_flag: false
# zero3_save_16bit_model: true
# zero_stage: 3
#distributed_type: stard
#downcast_bf16: 'no'
#dynamo_config: {}
#fsdp_config: {}
#machine_rank: 0
#main_training_function: main
#megatron_lm_config: {}
#mixed_precision: fp16
#num_machines: 2
#num_processes: 16
#tpu_env: []
#tpu_use_cluster: false
#tpu_use_sudo: false
#use_cpu: false