This commit is contained in:
Lewis Tunstall
2023-11-09 14:20:43 +00:00
parent 756bb76d22
commit 44b324487d
+1 -1
View File
@@ -28,7 +28,7 @@ max_prompt_length: 512
num_train_epochs: 3
optim: rmsprop
output_dir: data/zephyr-7b-dpo-full
per_device_train_batch_size: 4
per_device_train_batch_size: 8
per_device_eval_batch_size: 4
push_to_hub: true
save_strategy: "no"