PyTorch-VAE

Build a VAE model using PyTorch.

Environment settings

numpy==1.24.4
torch==2.4.1
torchvision==0.19.1
python==3.8.19

To run the model on CPU:

python -m src.train
python -m src.infer