/prototypical-representation-learning

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

Primary LanguagePython

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.