Former-commit-id: 450910c1db
This commit is contained in:
hiyouga
2023-06-27 23:54:24 +08:00
parent e877b8d55b
commit c5ec4eaef5
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