Skip to content

Interpretable Precise Year Prediction for Post-Medieval Western European Paintings

Notifications You must be signed in to change notification settings

randofan/vortex

Repository files navigation

VORTEX Painting-Year Prediction

End-to-end pipeline that mirrors the VORTEX paper (ViT + LoRA + CORAL).

# 1 env
conda create -n vortex python=3.10 -y
conda activate vortex
pip install -r requirements.txt

# 2 data → downloads & dedupes images
python scripts/download_images.py --csv data/paintings.csv --out data/images

# TODO run data split

# 3 train
python vortex/train.py --csv data/clean.csv --epochs 30

# 4 hyper-param tuning
python vortex/tune.py --csv data/clean.csv --random-trials 50 --timeout 3600

# TODO run test

TODOs

  • Check LoRa target module names in model.py:75
python - << 'EOF'
from transformers import AutoModel
model = AutoModel.from_pretrained("facebook/dinov2-giant")
# look for anything with query/key/value/proj/dense
names = [n for n,_ in model.named_modules() 
         if any(tok in n for tok in ("query","key","value","proj","dense"))]
print("\n".join(sorted(set(names))))
EOF

About

Interpretable Precise Year Prediction for Post-Medieval Western European Paintings

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •