BienKieu commited on
Commit
7d7f11c
·
verified ·
1 Parent(s): db44ab4

Upload final LLM2Seq training artifacts

Browse files
checkpoints/phase1_warmup_vlsp/config.yaml ADDED
@@ -0,0 +1,83 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adaptor:
2
+ encstack_layers: 4
3
+ fuse_layers:
4
+ - -1
5
+ - -4
6
+ - -8
7
+ - -12
8
+ fusion_type: tokenwise
9
+ num_global_memory_tokens: 16
10
+ type: gated_residual
11
+ use_encstack: true
12
+ use_global_memory_tokens: true
13
+ use_layer_fusion: true
14
+ use_salience_gate: true
15
+ data:
16
+ eval_file: llm2seq/data/processed/vlsp/val.jsonl
17
+ max_source_length: 8192
18
+ max_target_length: 768
19
+ source_prefix: "T\xF3m t\u1EAFt c\xE1c v\u0103n b\u1EA3n sau th\xE0nh m\u1ED9t \u0111\
20
+ o\u1EA1n ng\u1EAFn g\u1ECDn:\n"
21
+ test_file: llm2seq/data/processed/vlsp/test.jsonl
22
+ train_file: llm2seq/data/processed/vlsp/train.jsonl
23
+ evaluation:
24
+ eval_every_epochs: 1
25
+ metrics:
26
+ - rouge
27
+ - latency
28
+ save_every_epochs: 1
29
+ generation:
30
+ do_sample: false
31
+ eval_batch_size: 16
32
+ max_new_tokens: 768
33
+ min_new_tokens: 32
34
+ no_repeat_ngram_size: 3
35
+ repetition_penalty: 1.05
36
+ temperature: 0.0
37
+ top_k: 0
38
+ top_p: 1.0
39
+ huggingface:
40
+ enabled: true
41
+ fail_on_error: true
42
+ keep_local_epoch_checkpoints: 1
43
+ path_in_repo: checkpoints/phase1_warmup_vlsp
44
+ push_each_epoch: false
45
+ push_final_best: true
46
+ repo_id: null
47
+ repo_type: model
48
+ model:
49
+ d_dec: 1024
50
+ d_enc: 2048
51
+ encoder_name: McGill-NLP/LLM2Vec-Sheared-LLaMA-mntp
52
+ encoder_torch_dtype: bfloat16
53
+ encoder_trainable: false
54
+ project:
55
+ name: llm2seq
56
+ output_dir: runs/llm2seq_phase1_warmup_vlsp
57
+ task: summarization
58
+ small_decoder:
59
+ dropout: 0.1
60
+ ffn_size: 4096
61
+ hidden_size: 1024
62
+ max_seq_len: 1024
63
+ num_heads: 16
64
+ num_layers: 8
65
+ tie_embeddings: true
66
+ training:
67
+ adaptor_lr: 0.0005
68
+ batch_size: 2
69
+ bf16: true
70
+ decoder_lr: 0.0005
71
+ encoder_lr: 0.0
72
+ fp16: false
73
+ grad_accum_steps: 8
74
+ gradient_checkpointing: false
75
+ log_every_steps: 2
76
+ max_grad_norm: 1.0
77
+ min_lr_ratio: 0.1
78
+ num_train_epochs: 10
79
+ seed: 42
80
+ stage: phase1_warmup_vlsp
81
+ tf32: true
82
+ warmup_ratio: 0.05
83
+ weight_decay: 0.01