An open API service indexing awesome lists of open source software.

https://github.com/vahidzee/torchde

PyTorch wrapper for Deep Density Estimation Models
https://github.com/vahidzee/torchde

density-estimation energy-based-model masked-autoencoder pytorch sgld

Last synced: 8 months ago
JSON representation

PyTorch wrapper for Deep Density Estimation Models

Awesome Lists containing this project

README

          

# Torch Density Estimation
Generic implementation of Deep Neural Networks for Density Estimation in [PyTorch](https://pytorch.org/) and [PytorchLightning](https://www.pytorchlightning.ai/).

## Installation
In order to network implementations install `torchde` via pip:
```shell
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:

```shell
pip install "torchde[tools]"
```