Skip to content

Latest commit

 

History

History
19 lines (10 loc) · 787 Bytes

README.md

File metadata and controls

19 lines (10 loc) · 787 Bytes

Reproduction of Energy-based Model with Contrastive Divergence and Langevin Dynamics by Pytorch

the implementation of utils.py/Sampler refers to https://github.com/phlippe/uvadlc_notebooks/blob/master/docs/tutorial_notebooks/tutorial8/Deep_Energy_Models.ipynb

usage: python train.py

some sampling results after training 100 epochs:

imageData

loss trend:

energy of real samples are slightly lower than fake samples

image

contrastive divergence keeps fluctuating around 0 while regularization loss converges to 0 soon

image