MoritzWag / Representation-Learning

Representation Learning of Image Data with VAE.
2 stars 0 forks source link

Beta-Vae Implementation #16

Closed MoritzWag closed 4 years ago

MoritzWag commented 4 years ago

@alexpiehler implement beta Vae based on the following repo: https://github.com/AntixK/PyTorch-VAE betaVae: https://github.com/AntixK/PyTorch-VAE/blob/master/models/beta_vae.py The code structure between their and our repository is highly comparable. Pleas implement beta vae in file 'library/models2/beta_vae.py' and additional add the relevant config file as 'beta_vae.yaml'

work in branch: beta_vae

For doing this, follow these steps: 1.) git pull origin master 2.) git checkout -b beta_vae

and once your are done:

1.) git add * 2.) git commit 3.) git push origin beta_vae 4.) lastly create on github page a pull request (green button should pop-up) so that we can merge into master, once review is finished.