Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open
wants to merge 30 commits into
base: topic-recognition
Choose a base branch
from
Open

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

wants to merge 30 commits into from

Commits on Oct 11, 2022

  1. Initial commit and file setup

    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    7bcfbd0 View commit details
    Browse the repository at this point in the history

  2. Fixed issue where changes were showing up that I didn't make

    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    f331ab8 View commit details
    Browse the repository at this point in the history

  3. Added basic dataloader

    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    3805538 View commit details
    Browse the repository at this point in the history

  4. Changed the data import pipeline so it uses a pre-downloaded local fo… 

    …lder
    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    77dfc44 View commit details
    Browse the repository at this point in the history

  5. Fixed dataset loader

    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    7e9a6b5 View commit details
    Browse the repository at this point in the history

  6. Code cleanup

    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    343ba1b View commit details
    Browse the repository at this point in the history

  7. First implementation of VQVAE model architecture

    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    850e47b View commit details
    Browse the repository at this point in the history

  8. Implemented basic training loop and stats logging wrapper class for m… 

    …odel
    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    d8ada41 View commit details
    Browse the repository at this point in the history

  9. Changes to readme

    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    9456c9e View commit details
    Browse the repository at this point in the history

  10. Moved plotting function to utils.py

    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    b6bfc88 View commit details
    Browse the repository at this point in the history

  11. Added data preprocessing to the data loader

    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    4893b5c View commit details
    Browse the repository at this point in the history

  12. Added data variance to items returned by data loader

    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    a846d33 View commit details
    Browse the repository at this point in the history

  13. Verified training loop, added model saving functionality, fixed impor… 

    …ts bug
    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    72c9f06 View commit details
    Browse the repository at this point in the history

  14. Fully successfully implemented model saving and loading pipeline!!!

    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    e9ee4d9 View commit details
    Browse the repository at this point in the history

  15. Added exensive debug statements to aid in dataset debugging

    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    5e31873 View commit details
    Browse the repository at this point in the history

  16. Verified training script with oasis data and pulled hyperparameters u… 

    …p to top of training script for easy tuning
    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    3582b22 View commit details
    Browse the repository at this point in the history

  17. Added new todos to readme

    @Anwealso
    Anwealso committed Oct 11, 2022
    Configuration menu
    Copy the full SHA
    9e7dff3 View commit details
    Browse the repository at this point in the history

Commits on Oct 12, 2022

  1. Added model training metric over time plotting

    @Anwealso
    Anwealso committed Oct 12, 2022
    Configuration menu
    Copy the full SHA
    8d2878b View commit details
    Browse the repository at this point in the history

  2. Added example generation image logging throughout training to show pe… 

    …rformance progress
    @Anwealso
    Anwealso committed Oct 12, 2022
    Configuration menu
    Copy the full SHA
    c0d4762 View commit details
    Browse the repository at this point in the history

Commits on Oct 14, 2022

  1. Trained model up for 20 epochs, moved some plotting functions over to… 

    … utils
    @Anwealso
    Anwealso committed Oct 14, 2022
    Configuration menu
    Copy the full SHA
    db6fe74 View commit details
    Browse the repository at this point in the history

  2. Implemented average SSIM tracking through training and plotting over … 

    …time
    @Anwealso
    Anwealso committed Oct 14, 2022
    Configuration menu
    Copy the full SHA
    a5ca629 View commit details
    Browse the repository at this point in the history

  3. Updated and tweaked layout of readme

    @Anwealso
    Anwealso committed Oct 14, 2022
    Configuration menu
    Copy the full SHA
    446d940 View commit details
    Browse the repository at this point in the history

  4. Fixed up data importing on train.py, updated predict.py to actually run

    @Anwealso
    Anwealso committed Oct 14, 2022
    Configuration menu
    Copy the full SHA
    2019d37 View commit details
    Browse the repository at this point in the history

  5. Documentation updates and code cleanup

    @Anwealso
    Anwealso committed Oct 14, 2022
    Configuration menu
    Copy the full SHA
    16c7243 View commit details
    Browse the repository at this point in the history

Commits on Oct 20, 2022

  1. Implemented pixel cnn training loop (OOM error on local cpu so moving… 

    … to hpc)
    @Anwealso
    Anwealso committed Oct 20, 2022
    Configuration menu
    Copy the full SHA
    876e59a View commit details
    Browse the repository at this point in the history

Commits on Oct 21, 2022

  1. Trained vae for 20 epochs and pixelcnn for 60 epochs and created a ne… 

    …w set of oputput generations
    @Anwealso
    Anwealso committed Oct 21, 2022
    Configuration menu
    Copy the full SHA
    adeea63 View commit details
    Browse the repository at this point in the history

  2. Minor documentation tweaks

    @Anwealso
    Anwealso committed Oct 21, 2022
    Configuration menu
    Copy the full SHA
    9fc7920 View commit details
    Browse the repository at this point in the history

  3. Moved generator training over into train.py from generator_train.py

    @Anwealso
    Anwealso committed Oct 21, 2022
    Configuration menu
    Copy the full SHA
    8488064 View commit details
    Browse the repository at this point in the history

  4. Swapped epoch numbers back from test to production values

    @Anwealso
    Anwealso committed Oct 21, 2022
    Configuration menu
    Copy the full SHA
    8e90849 View commit details
    Browse the repository at this point in the history

  5. Removed some files that shouldn't be committed

    @Anwealso
    Anwealso committed Oct 21, 2022
    Configuration menu
    Copy the full SHA
    739d392 View commit details
    Browse the repository at this point in the history