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.