diff options
| -rwxr-xr-x | finetuning.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/finetuning.py b/finetuning.py index 161fb54..abb4b89 100755 --- a/finetuning.py +++ b/finetuning.py @@ -138,7 +138,7 @@ def main(): optim="adamw_torch_8bit", # Alternative from BnB: paged_adamw_8bit remove_unused_columns=False, logging_steps=args.logging_steps, - save_strategy="epochs", + save_strategy="epoch", ) trainer = SFTTrainer( |