Ecosyste.ms: Awesome

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

Awesome Lists | Featured Topics | Projects

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: 2 days 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]"
```