#!/bin/bash
# Full training pipeline on GPU instance
set -e

cd /workspace/arcisvlm

echo "=== Step 1: Download Flickr8k ==="
python scripts/download_flickr8k.py

echo ""
echo "=== Step 2: Train BPE Tokenizer ==="
python train_tokenizer.py

echo ""
echo "=== Step 3: Full Model Training ==="
python train.py

echo ""
echo "=== TRAINING COMPLETE ==="
ls -la checkpoints/