Skip to content

linzangsc/reproduction-of-EBM

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages