Skip to content

PyTorch wrapper for Deep Density Estimation Models

License

Notifications You must be signed in to change notification settings

vahidzee/torchde

Repository files navigation

Torch Density Estimation

Generic implementation of Deep Neural Networks for Density Estimation in PyTorch and PytorchLightning.

Installation

In order to network implementations install torchde via pip:

pip install torchde

In case you wish to employ the training procedures and datamodule implementations (on top of pytorch-lightning), you would need to install the additional tools as well:

pip install "torchde[tools]"