Skip to content

Latest commit

 

History

History
9 lines (6 loc) · 457 Bytes

README.md

File metadata and controls

9 lines (6 loc) · 457 Bytes

CS330-Project: Prototypes, Contrastive Learning and Classification under Novelty

Relevant Files

  • main_supcon_ood.py: Contains script for supervised contrastive training
  • protonet.py: Contains script for prototypical meta-training
  • experiments.ipyb: Contains code for t-SNE visualizations and model evaluations
  • data/cifardata.py: Defines data loaders for cifar10 and cifar100 dataset used for the various training and evaluation stages