Skip to content

README.md

Latest commit

 

History

History
16 lines (12 loc) · 613 Bytes

README.md

File metadata and controls

16 lines (12 loc) · 613 Bytes

sus_final_project

Final project for CS2952N with Prof. Chen Sun. Original repository linked here. Team effort with Nitya Thakkar and Dave Lubawski. Project report found in SUS_Embedding.pdf.

Workflow

  1. environment
  2. preprocess (msCOCO, MNIST, CIFAR-10)
  3. setup model
  4. train
  5. analysis

Notes and Guides

  • To run this code, you must download the MSCoco dataset into the ./data folder. You can see the file paths for this in preprocess.py. We downloaded MSCoco with wget.