Nexuss0781 commited on
Commit
b43566f
·
verified ·
1 Parent(s): 160c09f

Upload run_training.sh with huggingface_hub

Browse files
Files changed (1) hide show
  1. run_training.sh +64 -0
run_training.sh ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+ # Quick start script for training EthioBBPE tokenizer on Amharic Bible datasets
3
+
4
+ set -e
5
+
6
+ echo "=============================================="
7
+ echo "EthioBBPE Tokenizer Training - Quick Start"
8
+ echo "=============================================="
9
+
10
+ # Configuration
11
+ DATA_DIR="./data"
12
+ MODEL_NAME="EthioBBPE_AmharicBible"
13
+ VOCAB_SIZE=8000
14
+ COMPRESSION_FORMAT="gzip"
15
+ COMPRESSION_LEVEL=9
16
+ QUANTIZATION_BITS=8
17
+ MAX_CHECKPOINTS=3
18
+
19
+ echo ""
20
+ echo "Configuration:"
21
+ echo " Data Directory: $DATA_DIR"
22
+ echo " Model Name: $MODEL_NAME"
23
+ echo " Vocabulary Size: $VOCAB_SIZE"
24
+ echo " Compression: $COMPRESSION_FORMAT (level $COMPRESSION_LEVEL)"
25
+ echo " Quantization: ${QUANTIZATION_BITS}-bit"
26
+ echo " Max Checkpoints: $MAX_CHECKPOINTS"
27
+ echo ""
28
+
29
+ # Step 1: Prepare datasets (if not already done)
30
+ if [ ! -f "$DATA_DIR/combined_corpus.txt" ]; then
31
+ echo "Step 1: Preparing datasets..."
32
+ python scripts/prepare_datasets.py --output_dir $DATA_DIR
33
+ else
34
+ echo "Step 1: Dataset already prepared (combined_corpus.txt exists)"
35
+ fi
36
+
37
+ echo ""
38
+ echo "Step 2: Training tokenizer..."
39
+ python scripts/train_tokenizer.py \
40
+ --data_dir $DATA_DIR \
41
+ --vocab_size $VOCAB_SIZE \
42
+ --model_name $MODEL_NAME \
43
+ --compression_format $COMPRESSION_FORMAT \
44
+ --compression_level $COMPRESSION_LEVEL \
45
+ --enable_quantization \
46
+ --quantization_bits $QUANTIZATION_BITS \
47
+ --max_checkpoints $MAX_CHECKPOINTS \
48
+ --export_formats tokenizer.json vocab_compressed hf_export quantized \
49
+ --no_checkpoint
50
+
51
+ echo ""
52
+ echo "=============================================="
53
+ echo "Training Complete!"
54
+ echo "=============================================="
55
+ echo ""
56
+ echo "Model saved to: models/$MODEL_NAME/"
57
+ echo ""
58
+ echo "Files generated:"
59
+ ls -lh models/$MODEL_NAME/
60
+ echo ""
61
+ echo "Usage example:"
62
+ echo " from tokenizers import Tokenizer"
63
+ echo " tokenizer = Tokenizer.from_file('models/$MODEL_NAME/tokenizer.json')"
64
+ echo ""