File size: 1,326 Bytes
bfc6d2a | 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 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | #!/bin/bash
# Sync A1-Max MuQ LoRA checkpoints from Google Drive
# Run this before building the Docker image or uploading to HuggingFace
set -e
CHECKPOINT_DIR="./checkpoints"
GDRIVE_PATH="gdrive:crescendai_data/checkpoints/a1_max_sweep/A1max_r32_L7-12_ls0.1"
echo "A1-Max MuQ LoRA Checkpoint Sync"
echo "================================"
echo ""
echo "Creating checkpoint directories..."
mkdir -p "$CHECKPOINT_DIR/fold_0"
mkdir -p "$CHECKPOINT_DIR/fold_1"
mkdir -p "$CHECKPOINT_DIR/fold_2"
mkdir -p "$CHECKPOINT_DIR/fold_3"
echo ""
echo "Syncing A1-Max checkpoints (4-fold ensemble, 80.8% pairwise)..."
echo "Source: $GDRIVE_PATH"
echo ""
# Sync each fold's best checkpoint
for fold in 0 1 2 3; do
echo "Syncing fold_$fold..."
rclone copyto "$GDRIVE_PATH/fold_${fold}/best.ckpt" "$CHECKPOINT_DIR/fold_$fold/best.ckpt" --progress
done
echo ""
echo "Checkpoint sync complete!"
echo ""
echo "Directory structure:"
ls -la "$CHECKPOINT_DIR"
echo ""
for fold in 0 1 2 3; do
echo "fold_$fold:"
ls -la "$CHECKPOINT_DIR/fold_$fold"
done
echo ""
echo "Expected HuggingFace repository structure:"
echo " checkpoints/"
echo " fold_0/best.ckpt"
echo " fold_1/best.ckpt"
echo " fold_2/best.ckpt"
echo " fold_3/best.ckpt"
echo ""
echo "Model: A1-Max MuQ LoRA r32 L7-12 (6-dim, 80.8% pairwise, R2=0.50)"
|