Skip to content

Commit 3e4f747

Browse files
authored
Fix problem in math_mix config (agentscope-ai#196)
Co-authored-by: 问昊 <[email protected]>
1 parent 4f8cb4c commit 3e4f747

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/mix_math/mix_math.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@ algorithm:
1111
clip_range: 0.2
1212
use_token_level_loss_in_sft: true
1313
use_dynamic_bsz: true
14-
repeat_times: 8
1514
ppo_mini_batch_size: 320
1615
ppo_micro_batch_size_per_gpu: 4
1716
ngpus_trainer: 4
@@ -25,7 +24,7 @@ cluster:
2524
node_num: 1
2625
gpu_per_node: 8
2726
buffer:
28-
total_epochs: 1
27+
total_epochs: 10
2928
batch_size: 40
3029
max_retry_times: 3
3130
max_retry_interval: 1
@@ -56,6 +55,7 @@ buffer:
5655
storage_type: queue
5756
path: /PATH/TO/BUFFER/
5857
sft_warmup_dataset:
58+
total_epochs: 10
5959
name: math_sft
6060
storage_type: file
6161
algorithm_type: sft

0 commit comments

Comments
 (0)