quick_test.sh

#!/bin/bash
# 快速测试脚本 - 验证环境和模型

echo "=========================================="
echo "Quick Test for PriorCLIP-BLIP"
echo "=========================================="

# 激活环境
source /data/yy/miniconda3/etc/profile.d.sh
conda activate testPIR

cd /home/yy/testPIR

echo ""
echo "[Step 1/4] Verifying environment..."
python -c "
import torch
import transformers
print('✓ PyTorch:', torch.__version__)
print('✓ CUDA available:', torch.cuda.is_available())
print('✓ GPU count:', torch.cuda.device_count())
print('✓ Transformers:', transformers.__version__)
"

echo ""
echo "[Step 2/4] Testing BLIP adapter..."
python test_blip_mock.py

echo ""
echo "[Step 3/4] Testing with single tar file..."
python -m training.main \
    --model PriorCLIP_BLIP \
    --train-data '/data/yy/RS5M/train/pub11-train-0000.tar' \
    --dataset-type webdataset \
    --train-num-samples 1000 \
    --batch-size 4 \
    --epochs 1 \
    --lr 5e-4 \
    --warmup 10 \
    --precision fp32 \
    --workers 2 \
    --logs ./logs/quick_test \
    --name "QuickTest_${TIMESTAMP}"

echo ""
echo "[Step 4/4] Quick test completed!"
echo "If all tests passed, you can run full training:"
echo "  bash train_rs5m_3gpu.sh"