This commit is contained in:
hiyouga
2023-06-27 23:54:24 +08:00
parent 18f87c1b25
commit 450910c1db
4 changed files with 15 additions and 5 deletions

View File

@@ -208,6 +208,7 @@ accelerate launch src/train_XX.py # arguments (same as above)
compute_environment: LOCAL_MACHINE
deepspeed_config:
gradient_accumulation_steps: 4
gradient_clipping: 0.5
offload_optimizer_device: none
offload_param_device: none
zero3_init_flag: false