/FoVAE

2023 Edinburgh Independent Project -- a Foveating VAE with a Patch-Based Reconstruction Loss

Primary LanguageJupyter Notebook

FoVAE: Foveated VAE

Decent README pending

Training with default arguments (on MPS):

See default_configs/ for the default configs, which you can override.

python main.py fit --config experiment_overrides.yaml