Coding-With-Bashir commited on
Commit
be63b4f
·
verified ·
1 Parent(s): f1a3c50

Upload .\configs\test_small.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. .//configs//test_small.yaml +105 -0
.//configs//test_small.yaml ADDED
@@ -0,0 +1,105 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # BwengeAi Small Test Config
2
+ # Only 4 small datasets for end-to-end validation
3
+
4
+ project:
5
+ name: "BwengeAi"
6
+ version: "0.1.0"
7
+ description: "Rwandan AI that knows Kinyarwanda"
8
+
9
+ data:
10
+ raw_dir: "data/raw_test"
11
+ processed_dir: "data/processed_test"
12
+ huggingface:
13
+ datasets:
14
+ - name: "mbazaNLP/kinyarwanda_monolingual_v01.0"
15
+ split: "train"
16
+ description: "78.7K rows. Small Kinyarwanda monolingual corpus."
17
+ - name: "saillab/alpaca-kinyarwanda-cleaned"
18
+ split: "train"
19
+ description: "52K rows. Alpaca instruction tuning."
20
+ - name: "mbazaNLP/Kinyarwanda_English_parallel_dataset"
21
+ split: "train"
22
+ description: "55.7K rows. KR-EN parallel pairs."
23
+ - name: "masakhane/afrisenti"
24
+ split: "train"
25
+ config_name: "kin"
26
+ description: "5K rows. Kinyarwanda sentiment."
27
+ languages: ["rw"]
28
+
29
+ wikipedia:
30
+ api_url: "https://rw.wikipedia.org/w/api.php"
31
+ dump_url: "https://dumps.wikimedia.org/rwwiki/latest/"
32
+ dump_file: "rwwiki-latest-pages-articles.xml.bz2"
33
+
34
+ igihe:
35
+ base_url: "https://igihe.com"
36
+ categories: ["amakuru"]
37
+ rate_limit: 2.0
38
+ contact_email: "info@igihe.com"
39
+
40
+ kigalitoday:
41
+ base_url: "https://www.kigalitoday.com"
42
+ categories: ["amakuru"]
43
+ rate_limit: 2.0
44
+
45
+ newtimes:
46
+ rss_feeds:
47
+ - name: "News"
48
+ url: "https://www.newtimes.co.rw/rss"
49
+ rate_limit: 1.0
50
+
51
+ wikisource:
52
+ base_url: "https://rw.wikisource.org"
53
+ api_url: "https://rw.wikisource.org/w/api.php"
54
+
55
+ wiktionary:
56
+ api_url: "https://rw.wiktionary.org/w/api.php"
57
+
58
+ rbc:
59
+ base_url: "https://rbc.gov.rw"
60
+ rate_limit: 2.0
61
+
62
+ model:
63
+ base_model: "meta-llama/Llama-3.2-1B"
64
+ max_length: 512
65
+ vocab_size: 128256
66
+ num_hidden_layers: 16
67
+ num_attention_heads: 32
68
+ hidden_size: 2048
69
+ intermediate_size: 5632
70
+ dropout: 0.1
71
+
72
+ finetune:
73
+ base_models:
74
+ - "almanach/Llama-3-8B-mono-Kinyarwanda"
75
+ lora:
76
+ r: 16
77
+ lora_alpha: 32
78
+ lora_dropout: 0.05
79
+ target_modules: ["q_proj", "k_proj", "v_proj", "o_proj", "gate_proj", "up_proj", "down_proj"]
80
+
81
+ training:
82
+ output_dir: "outputs_test"
83
+ num_epochs: 1
84
+ batch_size: 2
85
+ gradient_accumulation_steps: 2
86
+ learning_rate: 2.0e-5
87
+ weight_decay: 0.01
88
+ warmup_steps: 10
89
+ max_grad_norm: 1.0
90
+ fp16: true
91
+ logging_steps: 5
92
+ save_steps: 100
93
+ eval_steps: 100
94
+ save_total_limit: 1
95
+ report_to: "tensorboard"
96
+ seed: 42
97
+
98
+ evaluation:
99
+ metrics:
100
+ - "perplexity"
101
+ - "rouge"
102
+ - "bleu"
103
+ - "accuracy"
104
+ test_size: 0.1
105
+ max_eval_samples: 100