forked from olivkoch/TinyRecursiveModels
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtrain_math_gsmk8.sh
More file actions
executable file
·27 lines (20 loc) · 900 Bytes
/
Copy pathtrain_math_gsmk8.sh
File metadata and controls
executable file
·27 lines (20 loc) · 900 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
#!/bin/bash
# TRM Mathematical Reasoning Training Script
# Trains TRM on MATH and GSM8K style mathematical reasoning problems
echo "🚀 Starting TRM Mathematical Reasoning Training"
echo "=============================================="
# Set environment variables
export DISABLE_COMPILE=1 # Disable torch.compile to avoid compilation issues
# Change to project directory
cd /home/anto/TinyRecursiveModels
# Run training with math config
echo "📚 Training on MATH & GSM8K dataset..."
echo "💾 Checkpoints will be saved to: checkpoints/TRM-Math-Reasoning/"
echo "📊 Training progress can be monitored via wandb (if enabled)"
echo ""
# Execute training
uv run python3 pretrain.py --config-name cfg_math_pretrain
echo ""
echo "✅ Training completed!"
echo "📁 Check the checkpoints directory for saved models"
echo "🧪 Use evaluate_math.py to test mathematical reasoning capabilities"