Topic-Recognition, VQ VAE (OASIS Dataset) - #461

Open

Anwealso wants to merge 30 commits into

shakes76:topic-recognitionfrom

Anwealso:topic-recognition

Commits on Oct 11, 2022

Initial commit and file setup
Anwealso
committed
7bcfbd0
Fixed issue where changes were showing up that I didn't make
Anwealso
committed
f331ab8
Added basic dataloader
Anwealso
committed
3805538
Changed the data import pipeline so it uses a pre-downloaded local folder
Anwealso
committed
77dfc44
Fixed dataset loader
Anwealso
committed
7e9a6b5
Code cleanup
Anwealso
committed
343ba1b
First implementation of VQVAE model architecture
Anwealso
committed
850e47b
Implemented basic training loop and stats logging wrapper class for model
Anwealso
committed
d8ada41
Changes to readme
Anwealso
committed
9456c9e
Moved plotting function to utils.py
Anwealso
committed
b6bfc88
Added data preprocessing to the data loader
Anwealso
committed
4893b5c
Added data variance to items returned by data loader
Anwealso
committed
a846d33
Verified training loop, added model saving functionality, fixed imports bug
Anwealso
committed
72c9f06
Fully successfully implemented model saving and loading pipeline!!!
Anwealso
committed
e9ee4d9
Added exensive debug statements to aid in dataset debugging
Anwealso
committed
5e31873
Verified training script with oasis data and pulled hyperparameters up to top of training script for easy tuning
Anwealso
committed
3582b22
Added new todos to readme
Anwealso
committed
9e7dff3

Commits on Oct 12, 2022

Commits on Oct 20, 2022

Implemented pixel cnn training loop (OOM error on local cpu so moving to hpc)
Anwealso
committed
876e59a