rosikand / prototypical-representation-learning

CS 231N Project: Protoypical Pre-training for Visual Representation Learning

Prototypical Representation Learning

Code for CS 231N project: "Prototypical Pre-Training for Visual Representation Learning"

Directory structure

• 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.