| # Fine-tuning script | |
| set -e | |
| # Configuration | |
| TRAINING_SET="${1:-data/processed/training_set.pkl}" | |
| MODEL_NAME="${2:-depth-anything/DA3-LARGE}" | |
| EPOCHS="${3:-10}" | |
| LR="${4:-1e-5}" | |
| CHECKPOINT_DIR="${5:-checkpoints}" | |
| echo "Starting fine-tuning..." | |
| echo "Training set: $TRAINING_SET" | |
| echo "Model: $MODEL_NAME" | |
| echo "Epochs: $EPOCHS" | |
| echo "Learning rate: $LR" | |
| echo "Checkpoint dir: $CHECKPOINT_DIR" | |
| # Create checkpoint directory | |
| mkdir -p "$CHECKPOINT_DIR" | |
| # Run fine-tuning | |
| python -m ylff.cli train \ | |
| --training-set "$TRAINING_SET" \ | |
| --model-name "$MODEL_NAME" \ | |
| --epochs "$EPOCHS" \ | |
| --lr "$LR" \ | |
| --checkpoint-dir "$CHECKPOINT_DIR" | |
| echo "Fine-tuning complete!" | |
| echo "Checkpoints saved to: $CHECKPOINT_DIR" | |