algorythmtechnologies commited on
Commit
328e2f5
·
verified ·
1 Parent(s): b49c004

Update run.zsh

Browse files
Files changed (1) hide show
  1. run.zsh +9 -12
run.zsh CHANGED
@@ -1,26 +1,24 @@
1
  #!/bin/zsh
2
  set -e
3
 
4
- # Activate environment if using venv or conda
5
- # source venv/bin/activate
6
 
7
- echo "🚀 Starting Zenith fine-tuning..."
8
-
9
- # Environment variables
10
  export BASE_MODEL="DeepSeek-Coder-V2-Lite-Instruct"
11
  export OUTPUT_DIR="outputs/zenith-lora"
12
  export DATA_PATH="data/zenith_combined.jsonl"
13
- export VAL_PATH=""
14
- export STEPS=200
15
  export BATCH=2
16
  export GRAD_ACC=2
17
- export LR=1e-4
18
- export WARMUP_RATIO=0.05
 
 
 
19
  export WEIGHT_DECAY=0.01
20
  export MAX_GRAD_NORM=1.0
21
  export LOG_STEPS=10
22
  export SAVE_STEPS=50
23
- export SAVE_LIMIT=3
24
  export EVAL_STEPS=50
25
  export EARLY_STOP_PATIENCE=3
26
  export SEED=42
@@ -28,7 +26,6 @@ export MAX_SEQ_LEN=2048
28
  export USE_4BIT=1
29
  export REPORT_TO="none"
30
 
31
- # Run training
32
  python3 train.py | tee outputs/train_log.txt
33
 
34
- echo "✅ Training finished! Check logs at outputs/train_log.txt"
 
1
  #!/bin/zsh
2
  set -e
3
 
4
+ echo "🚀 Starting Zenith fine-tuning (safe 2-hour config)..."
 
5
 
 
 
 
6
  export BASE_MODEL="DeepSeek-Coder-V2-Lite-Instruct"
7
  export OUTPUT_DIR="outputs/zenith-lora"
8
  export DATA_PATH="data/zenith_combined.jsonl"
9
+ export STEPS=300
 
10
  export BATCH=2
11
  export GRAD_ACC=2
12
+ export LR=5e-5
13
+ export LORA_R=8
14
+ export LORA_ALPHA=16
15
+ export LORA_DROPOUT=0.1
16
+ export WARMUP_RATIO=0.1
17
  export WEIGHT_DECAY=0.01
18
  export MAX_GRAD_NORM=1.0
19
  export LOG_STEPS=10
20
  export SAVE_STEPS=50
21
+ export SAVE_LIMIT=2
22
  export EVAL_STEPS=50
23
  export EARLY_STOP_PATIENCE=3
24
  export SEED=42
 
26
  export USE_4BIT=1
27
  export REPORT_TO="none"
28
 
 
29
  python3 train.py | tee outputs/train_log.txt
30
 
31
+ echo "✅ Training finished! Logs saved to outputs/train_log.txt"