Upload run_training.sh with huggingface_hub
Browse files- 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 ""
|