Create zero2_bf16_config.yaml

sunnweiwei · Oct 31, 2023 · 91593f0 · 91593f0
1 parent b6bd660
commit 91593f0
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/InstructDistill/zero2_bf16_config.yaml b/InstructDistill/zero2_bf16_config.yaml
@@ -0,0 +1,22 @@
+compute_environment: LOCAL_MACHINE
+debug: false
+deepspeed_config:
+  gradient_accumulation_steps: 1
+  gradient_clipping: 1.0
+  offload_optimizer_device: none
+  offload_param_device: none
+  zero3_init_flag: true
+  zero_stage: 2
+distributed_type: DEEPSPEED
+downcast_bf16: 'no'
+machine_rank: 0
+main_training_function: main
+mixed_precision: bf16
+num_machines: 1
+num_processes: 4
+rdzv_backend: static
+same_network: true
+tpu_env: []
+tpu_use_cluster: false
+tpu_use_sudo: false
+use_cpu: false