File size: 477 Bytes
00db46c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
# Default training configuration for SFT

# Learning parameters
learning_rate: 2e-5
weight_decay: 0.01
warmup_ratio: 0.1
lr_scheduler_type: linear
optim: adamw_torch

# Training parameters
num_train_epochs: 5
per_device_train_batch_size: 8
gradient_accumulation_steps: 8
bf16: true
remove_unused_columns: false

# SFT-specific parameters
max_length: 4096

# Logging and saving
report_to: tensorboard
logging_steps: 1
save_strategy: steps
save_steps: 10