ssl_retrieval trains retrieval representations for the 64×64 unlabeled image dataset. It contains frozen DINO inference, SimCLR, an autoencoder pretext task, four-way rotation prediction, and joint training. Retrieval always uses the normalized encoder embedding and cosine similarity.
python -m ssl_retrieval train --method simclr --output checkpoints/simclr.pt
python -m ssl_retrieval train --method joint --output… See the full description on the dataset page:
https://huggingface.co/datasets/duyle2408/milk10k_new_code.
