In this repository we demonstrate that stochastic weight averaging (https://arxiv.org/abs/1803.05407) can be implemented for (Variational) Auto-Encoder.
For experiments with a regular autoencoder see autoencoder_demos.py.
miro-code/SWA-VAE
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|