Skip to content

PyTorch implementations of algorithms for density estimation

License

Notifications You must be signed in to change notification settings

ikostrikov/pytorch-flows

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

42 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pytorch-flows

A PyTorch implementations of Masked Autoregressive Flow and some other invertible transformations from Glow: Generative Flow with Invertible 1x1 Convolutions and Density estimation using Real NVP.

For MAF, I'm getting results similar to ones reported in the paper. GLOW requires some work.

Run

python main.py --dataset POWER

Available datasets are POWER, GAS, HEPMASS, MINIBONE and BSDS300. For the moment, I removed MNIST and CIFAR10 because I have plans to add pixel-based models later.

Datasets

The datasets are taken from the original MAF repository. Follow the instructions to get them.

Tests

Tests check invertibility, you can run them as

pytest flow_test.py

Releases

No releases published

Packages

No packages published

Languages