Skip to content

Commit 344817a

Browse files
authored
[CICD] Update aquila tp2-pp2 ci for evaluation (flagos-ai#1110)
### PR Category CICD ### PR Types Improvements ### PR Description Update aquila tp2_pp2 ci for evaluation, to detect and validate the evaluation-related functionalities of FlagScale.
1 parent acfef26 commit 344817a

3 files changed

Lines changed: 15 additions & 4 deletions

File tree

flagscale/train/megatron/training/training.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2931,6 +2931,7 @@ def evaluate(
29312931
verbose=False,
29322932
non_loss_data_func=None,
29332933
extra_valid_index=None,
2934+
eval_iters=None,
29342935
):
29352936
"""Evaluation."""
29362937
args = get_args()

tests/functional_tests/train/aquila/conf/train/tp2_pp2.yaml

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,3 @@
1-
defaults:
2-
- data
3-
41
system:
52
tensor_model_parallel_size: 2
63
pipeline_model_parallel_size: 2
@@ -45,6 +42,8 @@ model:
4542
weight_decay: 0.1
4643
clip_grad: 1.0
4744
train_iters: 10
45+
eval_iters: 1
46+
eval_interval: 5
4847
micro_batch_size: 4
4948
global_batch_size: 1024
5049
seed: 42
@@ -59,3 +58,14 @@ model:
5958
lr_warmup_samples: 0
6059
lr_warmup_fraction: 0.01
6160
lr_decay_style: cosine
61+
62+
data:
63+
data_path: /home/gitlab-runner/data/pile_wikipedia_demo/pile_wikipedia_demo
64+
split: 80, 10, 10
65+
tokenizer:
66+
legacy_tokenizer: true
67+
tokenizer_type: AquilaTokenizerFS
68+
vocab_file: ./examples/aquila/tokenizer/vocab.json
69+
merge_file: ./examples/aquila/tokenizer/merges.txt
70+
special_tokens_file: ./examples/aquila/tokenizer/special_tokens.txt
71+
vocab_size: 100008
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
{"lm loss:": {"values": [11.63682, 11.63587, 11.49153, 11.38188, 11.28651, 11.21933, 11.17029, 11.12899, 11.10917, 11.08776]}}
1+
{"lm loss:": {"values": [11.63839, 11.63632, 11.49138, 11.38161, 11.28878, 11.21583, 74297.01, 11.22827, 11.16404, 11.1232, 11.10253, 11.07966, 11.07366, 64451.08, 11.07513, 64546.05, 11.07297, 64406.66]}}

0 commit comments

Comments
 (0)