cpu offloading

This commit is contained in:
Mads Henrichsen
2023-12-31 12:02:29 +01:00
parent d6273188f0
commit c371d6b546
2 changed files with 43 additions and 1 deletions

View File

@@ -54,7 +54,7 @@ eval_table_size:
eval_table_max_new_tokens: 128
saves_per_epoch: 1
debug:
deepspeed: deepspeed/zero3.json
deepspeed: deepspeed/zero3_cpu.json
weight_decay: 0.0
fsdp:
fsdp_config: