Build a VAE model using PyTorch.
numpy==1.24.4
torch==2.4.1
torchvision==0.19.1
python==3.8.19To run the model on CPU:
python -m src.train
python -m src.infer-
Notifications
You must be signed in to change notification settings - Fork 0
Build a VAE model using PyTorch.
License
Happme/PyTorch-VAE
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Build a VAE model using PyTorch.
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published