Code for CS 231N project: "Prototypical Pre-Training for Visual Representation Learning"
paper.pdf
: PDF of final project report.poster.pdf
: PDF of poster.logs/
: training logs for PRL fine-tuning.plots
: some saved t-SNE plots. See the report for a description.proto/
: directory containing the python scripts that implement prototypical pre-training and fine-tuning, as described in the report.simclr/
: directory containing scripts that implement SimCLR pre-training and fine-tuning.baseline/
: directory containing baseline (MLP and ResNet) script.