Former-commit-id: eed33862bc
This commit is contained in:
hiyouga
2024-06-03 19:12:29 +08:00
parent eaab09fccb
commit e4ce59243b
24 changed files with 52 additions and 52 deletions

View File

@@ -28,10 +28,10 @@ overwrite_output_dir: true
### train
per_device_train_batch_size: 1
gradient_accumulation_steps: 2
learning_rate: 0.0001
learning_rate: 1.0e-4
num_train_epochs: 3.0
lr_scheduler_type: cosine
warmup_steps: 0.1
warmup_ratio: 0.1
fp16: true
### eval