dh-mc commited on
Commit
5f306e7
·
1 Parent(s): 103dbc8

reduce b-->4

Browse files
llama-factory/config/qwen2_72b_lora_sft_4bit-p1.yaml CHANGED
@@ -28,7 +28,7 @@ overwrite_output_dir: true
28
  # resume_from_checkpoint: true
29
 
30
  ### train
31
- per_device_train_batch_size: 8
32
  gradient_accumulation_steps: 8
33
  learning_rate: 1.0e-4
34
  num_train_epochs: 3.0
 
28
  # resume_from_checkpoint: true
29
 
30
  ### train
31
+ per_device_train_batch_size: 4
32
  gradient_accumulation_steps: 8
33
  learning_rate: 1.0e-4
34
  num_train_epochs: 3.0
llama-factory/config/qwen2_72b_lora_sft_4bit-p2.yaml CHANGED
@@ -28,7 +28,7 @@ overwrite_output_dir: true
28
  # resume_from_checkpoint: true
29
 
30
  ### train
31
- per_device_train_batch_size: 8
32
  gradient_accumulation_steps: 8
33
  learning_rate: 1.0e-4
34
  num_train_epochs: 3.0
 
28
  # resume_from_checkpoint: true
29
 
30
  ### train
31
+ per_device_train_batch_size: 4
32
  gradient_accumulation_steps: 8
33
  learning_rate: 1.0e-4
34
  num_train_epochs: 3.0